Update app.py

app.py

@@ -3,33 +3,36 @@ import base64
 import requests
 import os
 
-# === CONFIGURATION ===
-
+# === GET API KEY FROM SECRETS ===
 GROQ_API_KEY = os.getenv("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")
 GROQ_MODEL = "llama3-70b-8192"
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
 
+# === DEBUG OUTPUT TO GRADIO UI ===
+def debug_env():
+    if GROQ_API_KEY is None:
+        return "❌ GROQ_API_KEY is NOT found in environment variables."
+    elif not GROQ_API_KEY.startswith("gsk_"):
+        return f"⚠️ GROQ_API_KEY is set, but seems invalid: {GROQ_API_KEY[:6]}..."
+    else:
+        return f"✅ GROQ_API_KEY loaded and starts correctly: {GROQ_API_KEY[:6]}..."
 
 # === IMAGE TO BASE64 ===
-
 def image_to_base64(image_path):
     try:
         with open(image_path, "rb") as image_file:
-            image_bytes = image_file.read()
-        return base64.b64encode(image_bytes).decode("utf-8")
-    except Exception as e:
+            return base64.b64encode(image_file.read()).decode("utf-8")
+    except Exception:
         return None
 
-
 # === PROMPT BUILDER ===
-
 def build_prompt(image_b64: str) -> str:
     return f"""
 You are an expert in Non-Destructive Testing (NDT) and industrial defect analysis.
 
-A user has uploaded an image of a defected component. Here's the base64 representation of that image (partial for size): {image_b64[:300]}...
+A user has uploaded an image of a defected component. Here's the base64 representation of that image (partial): {image_b64[:300]}...
 
-Analyze the defect and provide structured output with the following format:
+Analyze the defect and provide structured output:
 
 1. Defect Type:
 2. Recommended NDT Technique(s):
@@ -40,12 +43,12 @@ Analyze the defect and provide structured output with the following format:
 7. Preventive Measures:
 """
 
-
 # === GROQ API CALL ===
-
 def query_groq(prompt: str) -> str:
-    if not GROQ_API_KEY or not GROQ_API_KEY.startswith("gsk_"):
-        return "❌ Error: Invalid or missing Groq API key. Please check Space secrets."
+    if not GROQ_API_KEY:
+        return "❌ Error: GROQ_API_KEY is missing. Check Space secrets."
+    if not GROQ_API_KEY.startswith("gsk_"):
+        return "❌ Error: GROQ_API_KEY seems invalid. Should start with 'gsk_'."
 
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
@@ -65,39 +68,40 @@ def query_groq(prompt: str) -> str:
     try:
         response = requests.post(GROQ_API_URL, headers=headers, json=payload)
         if response.status_code == 401:
-            return "❌ Error 401: Invalid API Key. Please check your GROQ_API_KEY in Space secrets."
+            return "❌ Error 401: Invalid API Key. Check your Space secret."
         elif response.status_code != 200:
             return f"❌ Error {response.status_code}: {response.text}"
-
         return response.json()["choices"][0]["message"]["content"]
-
     except Exception as e:
-        return f"❌ Exception while calling Groq API: {str(e)}"
-
+        return f"❌ Exception: {str(e)}"
 
 # === MAIN ANALYSIS FUNCTION ===
-
 def analyze_defect(image_path):
     if not image_path:
-        return "Please upload an image."
+        return "⚠️ Please upload an image."
 
     image_b64 = image_to_base64(image_path)
     if not image_b64:
-        return "❌ Error reading image. Please try a different file."
+        return "❌ Failed to read the image. Try a different file."
 
     prompt = build_prompt(image_b64)
     return query_groq(prompt)
 
+# === GRADIO UI WITH DEBUG TAB ===
+with gr.Blocks() as demo:
+    gr.Markdown("# 🛠️ NDT Defect Analyzer")
 
-# === GRADIO UI ===
+    with gr.Tab("Analyze Image"):
+        image_input = gr.Image(type="filepath", label="Upload Defective Component Image")
+        output_box = gr.Textbox(label="Defect Analysis Report")
+        analyze_btn = gr.Button("Analyze")
+        analyze_btn.click(analyze_defect, inputs=image_input, outputs=output_box)
 
-demo = gr.Interface(
-    fn=analyze_defect,
-    inputs=gr.Image(type="filepath", label="Upload Defective Component Image"),
-    outputs=gr.Textbox(label="Defect Analysis Report"),
-    title="🛠️ NDT Defect Analyzer",
-    description="Upload an image of a damaged industrial component. The AI identifies the defect, recommends NDT method, and suggests solutions and preventive measures."
-)
+    with gr.Tab("🔍 Debug API Key"):
+        debug_output = gr.Textbox(label="Groq Key Status")
+        debug_btn = gr.Button("Check Key")
+        debug_btn.click(fn=debug_env, inputs=[], outputs=debug_output)
 
+# Run app
 if __name__ == "__main__":
     demo.launch()