Update app.py

app.py

@@ -98,8 +98,7 @@ def chatbot_response(history, user_input, text_input=None, file_input=None):
             # Append the question and answer to the chat history
             history.append([user_input, answer])
     
-    return history, ""
-
+    return history, answer
 
 # Streamlit App Layout
 st.set_page_config(page_title="News Classifier", page_icon="📰")
@@ -139,7 +138,7 @@ history = []
 user_input = st.text_input("Ask about news classification or topics", placeholder="Type a message...")
 source_toggle = st.radio("Select Context Source", ["Single Article", "Bulk Classification"])
 if st.button("✉ Send"):
-    history, bot_response = chatbot_response(history, user_input, source_toggle)
+    history, bot_response = chatbot_response(history, user_input, text_input=text_input if source_toggle == "Single Article" else None, file_input=file_input if source_toggle == "Bulk Classification" else None)
     st.write("*Chatbot Response:*")
     for q, a in history:
         st.write(f"*Q:* {q}")