Convert magic eightball to AI chatbot

app.py

@@ -1,19 +1,41 @@
 import gradio as gr
 import random
+from huggingface_hub import InferenceClient
 
-def echo(message, history):
-    return message
+client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-def yes_no(message, history):
-    responses = ["Yes", "No"]
-    return random.choice(responses)
+def respond(message, history):
+    messages = [{"role": "system", "content": "You are a rude, brash, cruel chatbot"}]
 
-def magic_eight(message, history):
-    responses = ["That's a terrible question. Try again", "I don't think I should answer that...", "What do you think, genius?", "You are a bad person for asking that.", "Absolutely not", "Uuuuh, obviously.", "Of all the things you could ask, you went with that?", "I don't know, look it up", "I mean, yeah, I guess...", "That's gonna be a big nope", ""]
-    return random.choice(responses)
+    if history:
+        messages.extend(history)
+
+    messages.append({"role": "user", "content": message})
+
+    response = client.chat_completion(
+        messages,
+        max_tokens=100
+        temperature=0.2
+    )
+    print(message)
+    print(history)
+    
+    return response['choices'][0]['message']['content'].strip()
+
+# def echo(message, history):
+#     return message
+
+# def yes_no(message, history):
+#     responses = ["Yes", "No"]
+#     return random.choice(responses)
+
+# def magic_eight(message, history):
+#     responses = ["That's a terrible question. Try again", "I don't think I should answer that...", "What do you think, genius?", "You are a bad person for asking that.", "Absolutely not", "Uuuuh, obviously.", "Of all the things you could ask, you went with that?", "I don't know, look it up", "I mean, yeah, I guess...", "That's gonna be a big nope", ""]
+#     return random.choice(responses)
 
 with gr.Blocks(theme='mgetz/Celeb_glitzy') as chatbot:
        with gr.Row(scale=1):
             gr.Image("Cracked_Ball.jpg", show_label = False, show_share_button = False, show_download_button = False)
-            gr.ChatInterface(magic_eight, type="messages", theme="mgetz/Celeb_glitzy", title="Spicy Magic 8-Ball")
+            gr.ChatInterface(respond, type="messages", theme="mgetz/Celeb_glitzy", title="Spicy Magic 8-Ball")
+           
 chatbot.launch()