Update app.py

app.py

@@ -45,24 +45,27 @@ st.write("Upload a file for classification:")
 uploaded_file = st.file_uploader("Choose a file", type=["csv", "pdf"])
 
 if uploaded_file is not None:
-    if uploaded_file.type == "text/csv":
-        # Process CSV file
-        df = pd.read_csv(uploaded_file)
-        if 'text' not in df.columns:
-            st.write("The CSV file must contain a 'text' column.")
-        else:
-            df['Prediction'] = df['text'].apply(lambda x: classify_text(x)[0])
-            df['Confidence'] = df['text'].apply(lambda x: classify_text(x)[1])
-            st.write(df)
+    try:
+        if uploaded_file.type == "text/csv":
+            # Process CSV file
+            df = pd.read_csv(uploaded_file, encoding='utf-8')
+            if 'text' not in df.columns:
+                st.write("The CSV file must contain a 'text' column.")
+            else:
+                df['Prediction'] = df['text'].apply(lambda x: classify_text(x)[0])
+                df['Confidence'] = df['text'].apply(lambda x: classify_text(x)[1])
+                st.write(df)
 
-    elif uploaded_file.type == "application/pdf":
-        # Process PDF file
-        with fitz.open(stream=uploaded_file.read(), filetype="pdf") as doc:
-            text = ""
-            for page in doc:
-                text += page.get_text()
+        elif uploaded_file.type == "application/pdf":
+            # Process PDF file
+            with fitz.open(stream=uploaded_file.read(), filetype="pdf") as doc:
+                text = ""
+                for page in doc:
+                    text += page.get_text()
 
-            # Perform classification
-            label, score = classify_text(text)
-            st.write(f"**Predicted Class for PDF:** {label}")
-            st.write(f"**Confidence:** {score:.4f}")
+                # Perform classification
+                label, score = classify_text(text)
+                st.write(f"**Predicted Class for PDF:** {label}")
+                st.write(f"**Confidence:** {score:.4f}")
+    except Exception as e:
+        st.error(f"Error: {e}")