SyedHasanCronosPMC commited on
Commit
6943499
·
verified ·
1 Parent(s): 93ad66c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +40 -0
app.py ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import requests
3
+ import gradio as gr
4
+
5
+ # Retrieve the API key from the environment variable
6
+ groq_api_key = os.getenv("GROQ_API_KEY")
7
+
8
+ if not groq_api_key:
9
+ raise ValueError("GROQ_API_KEY is missing! Set it in the Hugging Face Spaces 'Secrets'.")
10
+
11
+ # Define the API endpoint and headers
12
+ url = "https://api.groq.com/openai/v1/chat/completions"
13
+ headers = {"Authorization": f"Bearer {groq_api_key}"}
14
+
15
+ # Function to interact with Groq API
16
+ def chat_with_groq(user_input):
17
+ body = {
18
+ "model": "deepseek-r1-distill-qwen-32b",
19
+ "messages": [{"role": "user", "content": user_input}]
20
+ }
21
+
22
+ response = requests.post(url, headers=headers, json=body)
23
+
24
+ if response.status_code == 200:
25
+ return response.json()['choices'][0]['message']['content']
26
+ else:
27
+ return f"Error: {response.json()}"
28
+
29
+ # Create Gradio interface
30
+ interface = gr.Interface(
31
+ fn=chat_with_groq,
32
+ inputs=gr.Textbox(lines=2, placeholder="Ask me anything..."),
33
+ outputs=gr.Textbox(),
34
+ title="Chat with Groq AI (deepseek-r1-distill-qwen-32b)",
35
+ description="Type your question below and get a response powered by Groq's deepseek-r1-distill-qwen-32b model."
36
+ )
37
+
38
+ # Launch Gradio app
39
+ if __name__ == "__main__":
40
+ interface.launch()