Update app.py

app.py

@@ -1,7 +1,7 @@
 #import gradio as gr
 #gr.load("models/mistralai/Mistral-7B-Instruct-v0.3").launch()
 import os
-import gradio as gr
+import discord
 import requests
 from dotenv import load_dotenv
 
@@ -10,37 +10,36 @@ load_dotenv()
 
 API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
 headers = {"Authorization": f"Bearer {os.getenv('HFREAD')}"}
+DISCORD_TOKEN = os.getenv('dsTOK')
+
+client = discord.Client()
 
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
-def chatbot_response(input_text):
+def get_response(input_text):
     response = query({"inputs": input_text})
-    # Extract the generated text from the response
     if isinstance(response, dict) and 'generated_text' in response:
         return response['generated_text']
     elif isinstance(response, list) and len(response) > 0 and 'generated_text' in response[0]:
         return response[0]['generated_text']
     return 'No response generated.'
 
-# Gradio interface
-def main():
-    with gr.Blocks() as demo:
-        gr.Markdown("# Mistral-7B Chatbot")
-        
-        with gr.Row():
-            input_box = gr.Textbox(label="Input Text", placeholder="Type your question here...", lines=2)
-        
-        with gr.Row():
-            output_box = gr.Textbox(label="Response", placeholder="The response will appear here...", lines=5)
-        
-        submit_button = gr.Button("Submit")
-        
-        submit_button.click(fn=chatbot_response, inputs=input_box, outputs=output_box)
-    
-    demo.launch()
-
-if __name__ == "__main__":
-    main()
+@client.event
+async def on_ready():
+    print(f'We have logged in as {client.user}')
+
+@client.event
+async def on_message(message):
+    if message.author == client.user:
+        return
+
+    if message.content.startswith('!chat'):
+        input_text = message.content[len('!chat '):]
+        response_text = get_response(input_text)
+        await message.channel.send(response_text)
+
+client.run(DISCORD_TOKEN)
+