add groq

app.py

@@ -1,5 +1,12 @@
 import requests
 import gradio as gr
+from groq import Groq
+import os
+
+
+def initialize_groq_client():
+    """Initialize and return Groq client"""
+    return Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 def should_play(city):
     api_key = "bb5a3b68cd874753bf565453251804"
@@ -8,6 +15,19 @@ def should_play(city):
     res = requests.get(url, params=params).json()
     condition = res["current"]["condition"]["text"].lower()
     decision = "Don't Play Cricket" if "rain" in condition else "Go Play Cricket!"
-    return f"{city}: {condition.title()} → {decision}"
+
+    client = initialize_groq_client()
+    response = client.chat.completions.create(
+            model="llama-3.1-8b-instant",
+            messages=[
+                {"role": "system", "content": "You are an expert cricket coach."},
+                {"role": "user", "content": f"This is the weather condition {condition}, should we go to play cricket, answer 'Yes' or 'No'?"}
+            ],
+            temperature=0.7,
+            max_tokens=2000
+        )
+
+    return response.choices[0].message.content
+    # return f"{city}: {condition.title()} → {decision}"
 
 gr.Interface(fn=should_play, inputs="text", outputs="text", title="Cricket Decision App").launch()