Update app.py

app.py

@@ -158,17 +158,19 @@ def ask_llama(conversation_history, category, is_final_guess=False):
 # (replacing the original BlenderBot approach with DialoGPT for simplicity)
 def ask_help_agent(query):
     try:
-        from transformers import pipeline
-        # Using DialoGPT-medium as a free conversational text-generation model
-        help_chat = pipeline("text-generation", model="microsoft/DialoGPT-medium")
-        # The conversation string can incorporate previous messages if you want a dialogue
-        # Here we simply generate a response to the current query.
-        response = help_chat(query, max_length=100, do_sample=True, top_p=0.95)
-        # Trim the response to show only the generated text
-        return response[0]['generated_text']
+        from transformers import pipeline, Conversation
+        # Use the conversational pipeline with DialoGPT-medium for dialogue context
+        help_chat = pipeline("conversational", model="microsoft/DialoGPT-medium")
+        # Create a Conversation object to hold the context
+        conv = Conversation(query)
+        # Generate a response by passing the conversation object
+        response = help_chat(conv)
+        # Extract and return the generated response from the conversation
+        return conv.generated_responses[-1]
     except Exception as e:
         return f"Error in help agent: {str(e)}"
 
+
 # Main game logic
 def main():
     inject_custom_css()