Update app.py

app.py

@@ -1,44 +1,31 @@
 import gradio as gr
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
-# Load model and tokenizer
-tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-1B-distill")
-model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-1B-distill")
+# Load the model and tokenizer
+tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")  # Faster model
+model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
 
-ai_name = "Lan"  # Default AI name
-def chat_with_ai(user_input, history=[]):
+ai_name = "Lan"  # Default name
+
+# Chat function for Gradio
+def chat(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"{ai_name}: Wow! I love my new name: {ai_name}! 😍"
-    
-    # Handle name inquiry
+        return [(user_input, f"Wow! I love my new name: {ai_name}! 😍")]
+
     if "what is your name" in user_input.lower():
-        return f"{ai_name}: My name is {ai_name}! 😊"
-    
-    # Generate response
+        return [(user_input, f"My name is {ai_name}! 😊")]
+
+    # Tokenize input and 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 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()
+
+    return [(user_input, response)]
+
+# Gradio Interface
+iface = gr.ChatInterface(fn=chat, title="AI Friend Chatbot - Talk with Lan!", theme="dark")
+iface.launch()