Revert code and add openpyxl requirement

app.py

@@ -46,24 +46,30 @@ labels = st.text_input("Enter comma-separated labels:")
 
 if st.button("Calculate labels"):
 
-    labels_list = labels.split(",")
-    table = pd.read_excel(data)
-    table = table.loc[table["text"].apply(len) > 10].reset_index(drop=True)
+    try:
+        labels_list = labels.split(",")
+        table = pd.read_excel(data)
+        table = table.loc[table["text"].apply(len) > 10].reset_index(drop=True)
 
-    prog_bar = st.progress(0)
-    preds = []
+        prog_bar = st.progress(0)
+        preds = []
 
-    for i in range(len(table)):
-        preds.append(classifier(table.loc[i, "text"], labels)["labels"][0])
-        prog_bar.progress((i + 1) / len(table))
+        for i in range(len(table)):
+            preds.append(classifier(table.loc[i, "text"], labels)["labels"][0])
+            prog_bar.progress((i + 1) / len(table))
 
-    table["label"] = preds
+        table["label"] = preds
 
-    st.table(table[["text", "label"]])
+        st.table(table[["text", "label"]])
 
-    buf = BytesIO()
-    table[["text", "label"]].to_excel(buf)
+        buf = BytesIO()
+        table[["text", "label"]].to_excel(buf)
 
-    st.download_button(
-        label="Download table", data=buf.getvalue(), file_name="output.xlsx"
-    )
+        st.download_button(
+            label="Download table", data=buf.getvalue(), file_name="output.xlsx"
+        )
+
+    except:
+        st.error(
+            "Something went wrong. Make sure you upload an Excel file containing a column named `text` and a set of comma-separated labels is provided"
+        )

requirements.txt

@@ -5,4 +5,5 @@ sentencepiece
 xlrd
 -f https://download.pytorch.org/whl/torch_stable.html 
 torch
-appdirs
+appdirs
+openpyxl