Update app.py

app.py

@@ -41,36 +41,55 @@
 #     else:
 #         st.write(f"**Bot:** {message['content']}")
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
-st.title("🤖 Conversational Chatbot")
+st.title("🤖 Smart Chatbot")
 
 @st.cache_resource
-def load_chatbot():
-    return pipeline("conversational", model="facebook/blenderbot-400M-distill")
+def load_model():
+    model_name = "facebook/blenderbot-400M-distill"
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    return pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
-chatbot = load_chatbot()
+chatbot = load_model()
 
 if "conversation" not in st.session_state:
     st.session_state.conversation = []
 
-# Display history
+# Display chat history
 for msg in st.session_state.conversation:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
 
-if prompt := st.chat_input("Say something"):
+if prompt := st.chat_input("Ask me anything"):
     # Add user message
     st.session_state.conversation.append({"role": "user", "content": prompt})
     
-    # Generate response
-    with st.spinner("Thinking..."):
-        result = chatbot(str(st.session_state.conversation))
+    # Format context
+    context = "\n".join([f"{msg['role']}: {msg['content']}" for msg in st.session_state.conversation[-3:]])
     
-    # Extract bot response
-    response = result.generated_responses[-1]
-    
-    # Add to conversation
+    try:
+        with st.spinner("Thinking..."):
+            response = chatbot(
+                context,
+                max_length=200,
+                temperature=0.9,
+                top_k=60,
+                top_p=0.9,
+                num_beams=5,
+                no_repeat_ngram_size=3
+            )[0]['generated_text']
+            
+            # Clean response
+            response = response.split("assistant:")[-1].strip()
+            
+            # Ensure meaningful response
+            if not response or response.lower() in ["i don't know", "i'm not sure"]:
+                response = "I need to learn more about that. Could you clarify?"
+                
+    except Exception as e:
+        response = "Let me check my knowledge sources and get back to you on that."
+
     st.session_state.conversation.append({"role": "assistant", "content": response})
-    
     st.rerun()