Update app.py

app.py

@@ -10,13 +10,11 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 # Create generation pipeline
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-def respond(message, history):
+def respond(message):
     try:
-        # Combine the entire conversation history
-        input_text = message
-        if history:
-            input_text = "\n".join([f"User: {h[0]}\nBot: {h[1]}" for h in history]) + f"\nUser: {message}"
-        
+        # Only use the current message as input (no history)
+        input_text = f"User: {message}"
+
         # Generate response
         response = pipe(
             input_text,
@@ -28,17 +26,11 @@ def respond(message, history):
         )
         reply = response[0]['generated_text'].strip()
         
-        # Update history
-        history.append((message, reply))
-        return history
+        return reply
     
     except Exception as e:
         print(f"Error: {e}")
-        return history + [(message, "Sorry, something went wrong. Please try again.")]
-
-# Custom clear function
-def clear_history():
-    return []
+        return "Sorry, something went wrong. Please try again."
 
 # Set up Gradio chat interface
 with gr.Blocks() as demo:
@@ -47,10 +39,8 @@ with gr.Blocks() as demo:
     
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Your Message")
-    clear_btn = gr.Button("Clear History")
     
-    msg.submit(respond, [msg, chatbot], chatbot)
-    clear_btn.click(clear_history, None, chatbot, queue=False)
+    msg.submit(respond, msg, chatbot)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()