Update app.py

app.py

@@ -3,21 +3,19 @@ import base64
 import requests
 import os
 
-# === GET API KEY FROM SECRETS ===
+# ✅ LOAD API KEY from Space Secret
 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."
+        return "❌ GROQ_API_KEY is NOT found. Did you set the Space Secret and restart the Space?"
     elif not GROQ_API_KEY.startswith("gsk_"):
-        return f"⚠️ GROQ_API_KEY is set, but seems invalid: {GROQ_API_KEY[:6]}..."
+        return f"⚠️ GROQ_API_KEY is loaded but seems malformed: {GROQ_API_KEY[:6]}..."
     else:
-        return f"✅ GROQ_API_KEY loaded and starts correctly: {GROQ_API_KEY[:6]}..."
+        return f"✅ GROQ_API_KEY loaded: {GROQ_API_KEY[:6]}..."
 
-# === IMAGE TO BASE64 ===
 def image_to_base64(image_path):
     try:
         with open(image_path, "rb") as image_file:
@@ -25,25 +23,20 @@ def image_to_base64(image_path):
     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): {image_b64[:300]}...
-
-Analyze the defect and provide structured output:
-
-1. Defect Type:
-2. Recommended NDT Technique(s):
-3. Explanation of Defect Cause:
-4. Proposed Solution/Fix:
-5. Estimated Time to Repair:
-6. Tools Required:
-7. Preventive Measures:
+A user has uploaded an image of a defected component. Here's the base64 representation (partial): {image_b64[:300]}...
+Analyze and return the following:
+1. Defect Type
+2. Recommended NDT Technique(s)
+3. Explanation of Defect Cause
+4. Solution/Fix
+5. Time to Repair
+6. Tools Required
+7. Preventive Measures
 """
 
-# === GROQ API CALL ===
 def query_groq(prompt: str) -> str:
     if not GROQ_API_KEY:
         return "❌ Error: GROQ_API_KEY is missing. Check Space secrets."
@@ -68,40 +61,39 @@ 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. Check your Space secret."
+            return "❌ Error 401: Invalid API Key."
         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: {str(e)}"
 
-# === MAIN ANALYSIS FUNCTION ===
 def analyze_defect(image_path):
     if not image_path:
         return "⚠️ Please upload an image."
 
     image_b64 = image_to_base64(image_path)
     if not image_b64:
-        return "❌ Failed to read the image. Try a different file."
+        return "❌ Failed to read the image. Try again."
 
     prompt = build_prompt(image_b64)
     return query_groq(prompt)
 
-# === GRADIO UI WITH DEBUG TAB ===
+# ==== GRADIO UI ====
 with gr.Blocks() as demo:
-    gr.Markdown("# 🛠️ NDT Defect Analyzer")
+    gr.Markdown("# 🔍 NDT Defect Analyzer")
 
     with gr.Tab("Analyze Image"):
-        image_input = gr.Image(type="filepath", label="Upload Defective Component Image")
-        output_box = gr.Textbox(label="Defect Analysis Report")
+        image_input = gr.Image(type="filepath", label="Upload Image")
+        output_box = gr.Textbox(label="Analysis Output")
         analyze_btn = gr.Button("Analyze")
         analyze_btn.click(analyze_defect, inputs=image_input, outputs=output_box)
 
-    with gr.Tab("🔍 Debug API Key"):
-        debug_output = gr.Textbox(label="Groq Key Status")
-        debug_btn = gr.Button("Check Key")
+    with gr.Tab("Debug API Key"):
+        debug_output = gr.Textbox(label="API Key Debug")
+        debug_btn = gr.Button("Check")
         debug_btn.click(fn=debug_env, inputs=[], outputs=debug_output)
 
-# Run app
 if __name__ == "__main__":
     demo.launch()
+