Update app.py

app.py

@@ -149,6 +149,9 @@ with tab1:
             mime='text/csv'
         )
 
+        # Store classification results for Q&A pipeline
+        st.session_state['qa_df'] = df
+
 
 with tab2:
     st.header("Question Answering Pipeline")
@@ -156,10 +159,9 @@ with tab2:
     
     
     
-    if qa_file is not None:
-        qa_df = pd.read_csv(qa_file)
-        
-        context = ' '.join(qa_df['predicted_category'].tolist())
+    if 'qa_df' in st.session_state:
+        qa_df = st.session_state['qa_df']
+        context = ' '.join(qa_df['content'].tolist())  # Use predictions for Q&A
         st.write(f"Loaded {len(qa_df)} news excerpts")
 
     else: