Update app.py

app.py

@@ -7,8 +7,10 @@ tokenizer = AutoTokenizer.from_pretrained("MohamedTalaat91/gpt2-tokenizer")
 
 # Updated generate function
 def generate(messages, state):
-    input_text = messages[-1]  # The last message is a string
+    # The last message content is a dictionary with "role" and "content"
+    input_text = messages[-1]  # Extracting the content of the last message
     
+    # Tokenize the input
     inputs = tokenizer(input_text, return_tensors="pt")
     
     # Generate text based on the input
@@ -21,9 +23,14 @@ def generate(messages, state):
         temperature=0.7  # Controls randomness in sampling
     )
     
+    # Decode the generated text
     generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
     
-    return generated_text, state  # Return both generated text and state
+    # Prepare the response in the correct format
+    bot_message = {"role": "bot", "content": generated_text}
+    messages.append(bot_message)  # Add the bot's message to the conversation
+    
+    return messages, state  # Return the updated messages and state
 
 
 # Launch Gradio interface with the updated function