Update app.py

app.py

@@ -73,30 +73,16 @@ def classify_csv(file):
         return None, f"Error: {str(e)}"
 
 def chatbot_response(history, user_input, source):
+    # Always run the QA pipeline with the context
     user_input = user_input.lower()
     context = context_storage["context"] if source == "Single Article" else context_storage["bulk_context"]
-    num_articles = context_storage["num_articles"]
-    
-    if "number of articles" in user_input or "how many articles" in user_input:
-        answer = f"There are {num_articles} articles in the uploaded CSV."
-        history.append([user_input, answer])
-        return history, ""
     
+    # Run the QA pipeline and get the answer
     if context:
         result = qa_pipeline(question=user_input, context=context)
         answer = result["answer"]
         history.append([user_input, answer])
-        return history, ""
     
-    responses = {
-        "hello": "👋 Hello! How can I assist you with news today?",
-        "hi": "😊 Hi there! What do you want to know about news?",
-        "how are you": "🤖 I'm just a bot, but I'm here to help!",
-        "thank you": "🙏 You're welcome! Let me know if you need anything else.",
-        "news": "📰 I can classify news into Business, Sports, Politics, and more!",
-    }
-    response = responses.get(user_input, "🤔 I'm here to help with news classification and general info. Ask me about news topics!")
-    history.append([user_input, response])
     return history, ""
 
 # Streamlit App Layout