Update app.py

app.py

@@ -1,6 +1,6 @@
 # app.py
 import gradio as gr
-from chat import generate_response
+from chat import generate_response, reset_conversation
 
 def gradio_interface(user_input, max_tokens, temperature):
     return generate_response(user_input, max_length=max_tokens, temperature=temperature)
@@ -14,8 +14,24 @@ iface = gr.Interface(
     ],
     outputs="text",
     title="ChatGpt Tune",
-    description="Chat with your Hugging Face model using safetensors for fast loading!"
+    description="Chat with your Hugging Face model. Supports multi-turn conversation!"
 )
 
+# Optional: add a reset button for conversation memory
+def reset_button():
+    return reset_conversation()
+
+reset_iface = gr.Interface(
+    fn=reset_button,
+    inputs=[],
+    outputs="text",
+    live=False,
+    title="Reset Conversation",
+    description="Click to clear conversation memory."
+)
+
+# Combine interfaces into a single Gradio app
+demo = gr.TabbedInterface([iface, reset_iface], ["Chat", "Reset Conversation"])
+
 if __name__ == "__main__":
-    iface.launch(share=True)
+    demo.launch(share=True)