adding huggingface & transitioning to a generative AI model

app.py

@@ -1,16 +1,23 @@
-import random
 import gradio as gr
+from huggingface_hub import InferenceCLient
 
-def echo(message, history):
-    print(message)
-    print(history)
-    return message
+client = InterfaceClient("HuggingFaceH4/zephyr-7b-beta")
 
-def random_response(message, history):
-    return random.choice(["Yes", "No"])
+def respond(message, history):
+    messages = [{"role": "system", "content": "You are friend chatbot"}]\
 
-def magic_8_ball(message, history):
-    return random.choice(["It is certain", "Most likely", "Yes", "You may rely on it", "Ask again later", "Don't count on it", "My sources say no", "Cannot predict now", "Outlook not so good"])
+    if history:
+        messages.extend(history)
+
+    messages.append({"role": "user", "content": message})
+
+    response = client.chat_completion(
+        messages, 
+        max_tokens = 100, 
+        temperature = 0.2
+    )
+    return response['choices'][0]['message']['content'].strip()
+
+chatbot = gr.ChatInterface(, type = 'messages')
 
-chatbot = gr.ChatInterface(magic_8_ball, type = 'messages', title = 'Magic 8 Ball', description = "Ask me your most pressing questions!", theme = "mgetz/Celeb_glitzy")
 chatbot.launch()