Update app.py

app.py

@@ -2,39 +2,51 @@ import os
 import requests
 import gradio as gr
 
-# Load Groq API key from environment variable
-groq_api_key = os.getenv("GROQ_API_KEY")
+# Get Groq API key from environment variable
+groq_api_key = os.environ.get("GROQ_API_KEY")
+if not groq_api_key:
+    raise ValueError("Please set the GROQ_API_KEY in the Hugging Face Space secrets.")
 
-# Groq API endpoint
+# Groq API configuration
 url = "https://api.groq.com/openai/v1/chat/completions"
-
 headers = {
     "Authorization": f"Bearer {groq_api_key}"
 }
 
+# Prompt template
+template = """
+You are a friendly and professional customer service assistant for a telecom company.
+Respond to the customer's issue below with empathy and clear steps, especially for roaming support.
+
+Customer Query: {query}
+
+Your Response:
+"""
+
 # Function to call Groq API
-def chat_with_groq(user_input):
+def generate_response(user_query):
+    structured_prompt = template.format(query=user_query)
     body = {
         "model": "llama-3.1-8b-instant",
         "messages": [
-            {"role": "user", "content": user_input}
+            {
+                "role": "user",
+                "content": structured_prompt
+            }
         ]
     }
 
     response = requests.post(url, headers=headers, json=body)
-
     if response.status_code == 200:
         return response.json()['choices'][0]['message']['content']
     else:
-        return f"Error: {response.json()}"
+        return f"Error {response.status_code}: {response.text}"
 
 # Gradio interface
-interface = gr.Interface(
-    fn=chat_with_groq,
-    inputs=gr.Textbox(lines=2, placeholder="Ask me anything..."),
-    outputs=gr.Textbox(),
-    title="DDS Chat with Groq AI (Llama 3.1-8B)",
-    description="Ask your question to Groq's Llama 3.1-8B model."
-)
-
-interface.launch()
+gr.Interface(
+    fn=generate_response,
+    inputs=gr.Textbox(lines=4, placeholder="Describe your telecom issue..."),
+    outputs=gr.Textbox(label="Groq API Response"),
+    title="Telecom Support Assistant",
+    description="Ask your question and get a helpful reply from our AI-powered support assistant."
+).launch()