Update app.py

app.py

@@ -1,40 +1,44 @@
 import gradio as gr
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
-# Load the model and tokenizer
+# Load model and tokenizer
 tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-1B-distill")
 model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-1B-distill")
 
 ai_name = "Lan"  # Default AI name
-
-def chatbot_response(user_input):
+def chat_with_ai(user_input, history=[]):
     global ai_name
-
+    
     # Handle name change
     if "call you" in user_input.lower() or "your name is" in user_input.lower():
         new_name = user_input.split("call you")[-1].strip(" ?.!") or user_input.split("your name is")[-1].strip(" ?.!")
         ai_name = new_name.capitalize()
-        return f"Wow! I love my new name: {ai_name}! 😍"
-
+        return f"{ai_name}: Wow! I love my new name: {ai_name}! 😍"
+    
     # Handle name inquiry
     if "what is your name" in user_input.lower():
-        return f"My name is {ai_name}! 😊"
-
+        return f"{ai_name}: My name is {ai_name}! 😊"
+    
     # Generate response
     inputs = tokenizer(user_input, return_tensors="pt")
     reply_ids = model.generate(**inputs, max_length=100)
     response = tokenizer.decode(reply_ids[0], skip_special_tokens=True)
-
-    return response
-
-# Create Gradio interface
-iface = gr.Interface(
-    fn=chatbot_response,
-    inputs="text",
-    outputs="text",
-    title="AI Friend Chatbot",
-    description="Chat with your AI friend! Type a message below.",
-)
-
-if __name__ == "__main__":
-    iface.launch()
+    
+    return f"{ai_name}: {response}"
+
+# Gradio UI
+def chat_interface():
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+        gr.Markdown("## 🤖 AI Friend Chatbot - Talk with Lan!")
+        with gr.Row():
+            user_input = gr.Textbox(placeholder="Type your message...", label="You:")
+        chat_output = gr.Textbox(label="Lan:", interactive=False)
+        with gr.Row():
+            submit_btn = gr.Button("Send", variant="primary")
+            clear_btn = gr.Button("Clear")
+        
+        submit_btn.click(chat_with_ai, inputs=[user_input], outputs=[chat_output])
+        clear_btn.click(lambda: "", inputs=[], outputs=[chat_output])
+    return demo
+
+chat_interface().launch()