Update app.py

app.py

@@ -1,51 +1,54 @@
 #import gradio as gr
 #gr.load("models/mistralai/Mistral-7B-Instruct-v0.3").launch()
-
 import os
-import gradio as gr
 import requests
+import discord
+from discord.ext import commands
 from dotenv import load_dotenv
 
-import gradio_client as grc
-grc.Client("Z3ktrix/mistralai-Mistral-7B-Instruct-v0.3").deploy_discord(id={"Authorization": f"Bearer {os.getenv('DSTOK')}"})
-# Load the environment variables from the .env file
+# Load environment variables from the .env file
 load_dotenv()
 
-API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
-headers = {"Authorization": f"Bearer {os.getenv('HFREAD')}"}
-
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+DISCORD_TOKEN = os.getenv('DSTOK')
+HF_API_KEY = os.getenv('HFREAD')
 
-def chatbot_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']
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
+headers = {"Authorization": f"Bearer {HF_API_KEY}"}
+
+# Function to query the Hugging Face model
+def query_huggingface(payload):
+    try:
+        response = requests.post(API_URL, headers=headers, json=payload)
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        print(f"Error querying the API: {e}")
+        return {"error": str(e)}
+
+# Initialize the Discord bot
+intents = discord.Intents.default()
+intents.messages = True
+intents.message_content = True
+
+bot = commands.Bot(command_prefix="!", intents=intents)
+
+@bot.event
+async def on_ready():
+    print(f'Bot is ready. Logged in as {bot.user}')
+
+@bot.command(name='ask')
+async def ask(ctx, *, question: str):
+    """
+    Command to ask a question to the Hugging Face model.
+    """
+    await ctx.send(f"Question: {question}")
+    response = query_huggingface({"inputs": question})
+    if 'generated_text' in response:
+        await ctx.send(f"Response: {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("Chatty")
-        
-        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()
-    
+        await ctx.send(f"Response: {response[0]['generated_text']}")
+    else:
+        await ctx.send("Sorry, I couldn't generate a response.")
+
+# Run the bot
+bot.run(DISCORD_TOKEN)