Update app.py

app.py

@@ -1,17 +1,20 @@
 import gradio as gr
 from groq import Groq
 import os
+import time
 
-# Get Groq API key from environment
+# -------------------------
+# Initialize Groq client
+# -------------------------
 api_key = os.environ.get("GROQ_API_KEY")
-
-# Initialize Groq client only if key exists
 client = Groq(api_key=api_key) if api_key else None
 
+# -------------------------
+# Main recommendation function
+# -------------------------
 def get_recommendation(quantity, food_type, expiry_hours):
-    print("DEBUG: API key loaded?", os.environ.get("GROQ_API_KEY") is not None)
     if not client:
-        return "❌ GROQ_API_KEY not found. Please set your API key in environment variables or Secrets."
+        return "❌ GROQ_API_KEY not found. Please set your API key in Secrets."
 
     # Basic routing logic
     if quantity < 20:
@@ -32,20 +35,25 @@ The system selected: {base}
 Explain why this is the best recipient in 2-3 lines.
 """
 
-    # AI call
-    try:
-        response = client.chat.completions.create(
-            model="llama-2-7b-chat",
-            messages=[{"role": "user", "content": prompt}]
-        )
-        explanation = response.choices[0].message.content
-    except Exception as e:
-        explanation = f"❌ AI explanation not available:\n{e}"
+    # Retry logic for temporary connection issues
+    explanation = "❌ AI explanation not available."
+    for attempt in range(3):
+        try:
+            response = client.chat.completions.create(
+                model="llama-2-7b-chat",  # smaller model for Spaces
+                messages=[{"role": "user", "content": prompt}]
+            )
+            explanation = response.choices[0].message.content
+            break
+        except Exception as e:
+            explanation = f"❌ AI request failed on attempt {attempt+1}: {e}"
+            time.sleep(2)  # wait 2 seconds before retrying
 
     return f"✅ Recommended: {base}\n\n🤖 AI Reasoning:\n{explanation}"
 
-
+# -------------------------
 # Gradio Blocks UI
+# -------------------------
 with gr.Blocks(title="🍽️ NourishNet AI") as demo:
     gr.Markdown(
         """
@@ -61,7 +69,7 @@ with gr.Blocks(title="🍽️ NourishNet AI") as demo:
         food_type_input = gr.Dropdown(["halal", "soft", "packaged"], label="Food Type")
         expiry_input = gr.Number(label="Hours Until Expiry", value=5)
 
-    output_text = gr.Textbox(label="Recommendation & AI Reasoning")
+    output_text = gr.Textbox(label="Recommendation & AI Reasoning", lines=7)
 
     submit_btn = gr.Button("Get Recommendation")
     submit_btn.click(
@@ -70,5 +78,8 @@ with gr.Blocks(title="🍽️ NourishNet AI") as demo:
         outputs=output_text
     )
 
+# -------------------------
+# Run the app
+# -------------------------
 if __name__ == "__main__":
     demo.launch()