Update app.py

app.py

@@ -105,13 +105,8 @@ with tab1:
     uploaded_file = st.file_uploader("Choose a CSV file", type="csv")
 
 
-    # Load the CSV file
-
-
-
-    
         
-    # Check if 'excerpt' column exists
+    # Check the file
     if uploaded_file is None:
         st.warning("Please upload a CSV file.")
         
@@ -155,25 +150,19 @@ with tab1:
 
 
 with tab2:
-    '''st.header("Question Answering Pipeline")
-    st.write("Ask questions about news content and get answers from our AI model.")'''
-    
-    # Option to upload file or enter text manually
-    input_option = st.radio("Choose input method:", ("Upload CSV", "Enter Text Manually"))
+    st.header("Question Answering Pipeline")
+    st.write("Ask questions about news content and get answers from our AI model.")
     
-    context = ""
     
-    if input_option == "Upload CSV":
-        qa_file = st.file_uploader("Upload news content (CSV)", type="csv")
-        if qa_file is not None:
-            qa_df = pd.read_csv(qa_file)
-            if 'excerpt' not in qa_df.columns:
-                st.error("CSV must contain an 'excerpt' column")
-            else:
-                context = ' '.join(qa_df['excerpt'].tolist())
-                st.write(f"Loaded {len(qa_df)} news excerpts")
+    if qa_file is not None:
+        qa_df = uploaded_file
+
+        context = ' '.join(qa_df['predicted_category'].tolist())
+        st.write(f"Loaded {len(qa_df)} news excerpts")
+
     else:
-        context = st.text_area("Paste news content here:", height=200)
+        st.warning("Please upload a CSV file.")
+   
     
     question = st.text_input("Enter your question:")