Update app.py

app.py

@@ -1,82 +1,35 @@
-# 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, user_name):
-    messages = [{"role": "system", "content": f"You are a friendly chatbot talking to {user_name}."}]
-    
+user_profile = {
+    "name": ""
+}
+
+def respond(message, history):
+    messages = [{"role": "system", "content": "You are a friendly chatbot."}]
+
     if history:
         messages.extend(history)
-    
+    else:
+        messages.append({"role": "assistant", "content": f"Hi, {user_profile['name']}! How can I help you today?"})
+
     messages.append({"role": "user", "content": message})
-    
-    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)
+    response = client.chat_completion(
+        messages,
+        max_tokens=100
+    )
 
-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, ""
+    print(messages)
+    return response['choices'][0]['message']['content'].strip()
 
 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]
-    )
+    name = gr.Textbox(label="Name")
+    submit_btn = gr.Button("Submit")
+    submit_btn.click(respond, inputs=name, outputs=user_profile['name'])
+
+chatbot = gr.ChatInterface(respond, type="messages")
 
 chatbot.launch()