Update app.py

app.py

@@ -1,44 +1,37 @@
 import gradio as gr
-import requests
 import os
+from groq import Groq
 
 # Fetch the Groq API key from Gradio secrets
 GROQ_API_KEY = os.getenv("Groq_API_Key")
 
-# Groq's LLaMA2 endpoint
-GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
+# Initialize Groq client with the secret API key
+client = Groq(api_key=GROQ_API_KEY)
 
 # System prompt to define bot behavior
-system_prompt = (
+SYSTEM_PROMPT = (
     "You are a Construction Material Advisor Bot. Your job is to recommend the best construction materials "
     "for civil engineering projects based on environmental conditions, budget, load requirements, and durability. "
-    "Be concise, informative, and explain pros and cons of materials."
+    "Be concise, informative, and explain the pros and cons of the recommended materials."
 )
 
-# Function to get response from Groq API
-def get_material_advice(user_query):
-    headers = {
-        "Authorization": f"Bearer {GROQ_API_KEY}",
-        "Content-Type": "application/json"
-    }
-
-    payload = {
-        "model": "llama2-70b-4096",  # You can use llama2-7b if available
-        "messages": [
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": user_query}
-        ],
-        "temperature": 0.7,
-        "max_tokens": 500
-    }
+MODEL_NAME = "llama3-70b-8192"
 
+def get_material_advice(user_query: str) -> str:
+    """Return a material recommendation generated by the Groq LLM."""
     try:
-        response = requests.post(GROQ_API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        data = response.json()
-        return data["choices"][0]["message"]["content"].strip()
-    except Exception as e:
-        return f"Error: {str(e)}"
+        chat_completion = client.chat.completions.create(
+            model=MODEL_NAME,
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": user_query},
+            ],
+            temperature=0.7,
+            max_tokens=500,
+        )
+        return chat_completion.choices[0].message.content.strip()
+    except Exception as exc:
+        return f"Error: {exc}"
 
 # Gradio interface
 demo = gr.Interface(
@@ -46,8 +39,12 @@ demo = gr.Interface(
     inputs=gr.Textbox(lines=3, placeholder="Ask about construction materials..."),
     outputs="text",
     title="🧱 Construction Material Advisor Bot",
-    description="Ask questions like 'What is the best material for a coastal bridge?' or 'Suggest budget-friendly wall material for cold regions.'"
+    description=(
+        "Ask questions like 'What is the best material for a coastal bridge?' or "
+        "'Suggest budget-friendly wall material for cold regions.'"
+    ),
 )
 
-# Launch app
-demo.launch()
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()