Update app.py

app.py

@@ -9,12 +9,28 @@ os.environ["HF_HOME"] = HUGGINGFACE_API_KEY  # Set the Hugging Face API key
 # Initialize the text generation pipeline (without passing api_key)
 chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 
+# Initialize the conversation history
+if "history" not in st.session_state:
+    st.session_state["history"] = []
+
 # Function to get response from the model
 def get_chatbot_response(user_input):
     try:
+        # Prepare the conversation history for the model
+        conversation_history = ""
+        for user_input, response in st.session_state["history"]:
+            conversation_history += f"User: {user_input}\nBot: {response}\n"
+        
+        # Add the current user input to the conversation
+        conversation_history += f"User: {user_input}\n"
+        
         # Generate response from the model
-        response = chatbot(user_input)
-        return response[0]["generated_text"]
+        response = chatbot(conversation_history, max_length=1000, pad_token_id=50256)[0]["generated_text"]
+        
+        # Remove the user input from the generated response (optional)
+        response = response[len(conversation_history):].strip()
+        
+        return response
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -66,9 +82,6 @@ st.markdown("""
 st.markdown('<div class="chat-header">Gemini Chatbot-Your AI Companion 💻</div>', unsafe_allow_html=True)
 st.write("Powered by Hugging Face’s DialoGPT model for smart, engaging conversations. 🤖")
 
-if "history" not in st.session_state:
-    st.session_state["history"] = []
-
 with st.form(key="chat_form", clear_on_submit=True):
     user_input = st.text_input("Your message here... ✍️", max_chars=2000, label_visibility="collapsed")
     submit_button = st.form_submit_button("Send 🚀")