Update app.py

app.py

@@ -18,18 +18,18 @@ 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"]:
+        for user_input, response in st.session_state["history"][-5:]:  # Limit history to last 5 exchanges
             conversation_history += f"User: {user_input}\nBot: {response}\n"
         
         # Add the current user input to the conversation
         conversation_history += f"User: {user_input}\n"
         
         # Debug: Print the conversation history
-        print("Conversation History:")
+        print("Conversation History (Trimmed if Necessary):")
         print(conversation_history)
         
         # Generate response from the model
-        response = chatbot(conversation_history, max_length=1000, pad_token_id=50256)[0]["generated_text"]
+        response = chatbot(conversation_history, max_length=1000, pad_token_id=50256, num_return_sequences=1)[0]["generated_text"]
         
         # Debug: Print the full response generated by the model
         print("Generated Response (Before Stripping User Input):")