Update app.py

app.py

@@ -1,7 +1,8 @@
 #import gradio as gr
 #gr.load("models/mistralai/Mistral-7B-Instruct-v0.3").launch()
+
 import os
-import discord
+import gradio as gr
 import requests
 from dotenv import load_dotenv
 
@@ -10,36 +11,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 get_response(input_text):
+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']
     elif isinstance(response, list) and len(response) > 0 and 'generated_text' in response[0]:
         return response[0]['generated_text']
     return 'No response generated.'
 
-@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)
-
-
+# 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()