Update app.py

app.py

@@ -11,9 +11,16 @@ def chat(message, history):
     # Use the GPT-2 tokenizer for user input
     user_input = gpt2_tokenizer.encode(message, return_tensors="pt")
 
+    # Prepare conversation history as a list of dictionaries
+    conversation = [{"role": "system", "content": "You are a helpful assistant."},
+                    {"role": "user", "content": message}]
+    
+    if history:
+        conversation.append({"role": "assistant", "content": history})
+
     # Generate a response from DialoGPT-medium
-    response = chatbot(message)
-    bot_response = response[0]['message']
+    response = chatbot(conversation)
+    bot_response = response[0]['message']['content']
 
     return bot_response