Update app.py

app.py

@@ -3,7 +3,7 @@ from langchain_huggingface import HuggingFacePipeline
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 
-# Load a very lightweight model (distilgpt2, ~82M parameters, fast on CPU)
+# Load lightweight model (distilgpt2, ~82M parameters, fast on CPU)
 llm = HuggingFacePipeline.from_model_id(
     model_id="distilgpt2",
     task="text-generation",
@@ -26,6 +26,11 @@ conversation = ConversationChain(
     verbose=False  # No logging for speed
 )
 
+# Clear memory function (called via a separate button in the UI)
+def clear_memory():
+    memory.clear()
+    return "Conversation history cleared!"
+
 # Chat function
 def chat_with_agent(message, history):
     try:
@@ -37,19 +42,20 @@ def chat_with_agent(message, history):
     return response
 
 # Gradio chat interface
-iface = gr.ChatInterface(
-    fn=chat_with_agent,
-    title="Fast Free AI Agent",
-    description="A lightweight conversational AI that remembers our talks. Hosted free on Hugging Face Spaces. Responses in ~3-10 seconds.",
-    examples=["My name is Alex. What's my name?", "Tell me a short joke."]
-)
-
-# Add a clear memory button
-def clear_memory():
-    memory.clear()
-    return "Conversation history cleared!"
-
-iface.additional_inputs = [gr.Button("Clear Memory", onclick=clear_memory)]
-
+with gr.Blocks() as iface:
+    gr.Markdown("# Fast Free AI Agent")
+    gr.Markdown("A lightweight conversational AI that remembers our talks. Hosted free on Hugging Face Spaces. Responses in ~3-10 seconds.")
+    chatbot = gr.ChatInterface(
+        fn=chat_with_agent,
+        examples=[
+            {"text": "My name is Alex. What's my name?"},
+            {"text": "Tell me a short joke."}
+        ],
+        title="Chat with Your AI Agent",
+        description="Type your message below to chat. The AI remembers our conversation!"
+    )
+    gr.Button("Clear Conversation History").click(fn=clear_memory, outputs=gr.Textbox())
+
+# Launch the app
 if __name__ == "__main__":
     iface.launch()