Update app.py

app.py

@@ -1,24 +1,82 @@
+# import gradio as gr
+# from huggingface_hub import InferenceClient
+
+# client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+
+# def respond(message, history):
+#     messages = [{"role": "system", "content": "You are a friendly chatbot."}]
+
+#     if history:
+#         messages.extend(history)
+
+#     messages.append({"role": "user", "content": message})
+
+#     response = client.chat_completion(
+#         messages,
+#         max_tokens=100
+#     )
+
+#     return response['choices'][0]['message']['content'].strip()
+
+
+# chatbot = gr.ChatInterface(respond, type="messages")
+
+# chatbot.launch()
+
 import gradio as gr
 from huggingface_hub import InferenceClient
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-def respond(message, history):
-    messages = [{"role": "system", "content": "You are a friendly chatbot."}]
-
+def respond(message, history, user_name):
+    messages = [{"role": "system", "content": f"You are a friendly chatbot talking to {user_name}."}]
+    
     if history:
         messages.extend(history)
-
+    
     messages.append({"role": "user", "content": message})
-
-    response = client.chat_completion(
-        messages,
-        max_tokens=100
-    )
-
+    
+    response = client.chat_completion(messages, max_tokens=100)
     return response['choices'][0]['message']['content'].strip()
 
+def start_chat(name):
+    if name:
+        return f"Hello {name}! How can I help you today?", gr.update(interactive=True)
+    return "", gr.update(interactive=False)
+
+def chat_fn(message, history, user_name):
+    if not user_name:
+        return history, ""
+    
+    bot_response = respond(message, history, user_name)
+    history.append([message, bot_response])
+    return history, ""
 
-chatbot = gr.ChatInterface(respond, type="messages")
+with gr.Blocks() as chatbot:
+    gr.Markdown("# Personal Chatbot")
+    
+    user_name = gr.State("")
+    
+    with gr.Row():
+        with gr.Column():
+            name_input = gr.Textbox(label="Your Name")
+            start_btn = gr.Button("Start Chat")
+            
+        with gr.Column():
+            chatbot = gr.Chatbot(height=400)
+            msg_input = gr.Textbox(label="Message", interactive=False)
+            send_btn = gr.Button("Send")
+    
+    start_btn.click(
+        lambda name: [name, start_chat(name)[0], start_chat(name)[1]],
+        inputs=[name_input],
+        outputs=[user_name, chatbot, msg_input]
+    )
+    
+    send_btn.click(
+        chat_fn,
+        inputs=[msg_input, chatbot, user_name],
+        outputs=[chatbot, msg_input]
+    )
 
 chatbot.launch()