Update app.py

app.py

@@ -5,28 +5,30 @@ from PIL import Image
 import io
 import os
 
-# ✅ Read the Groq API key from environment variable (set this in Hugging Face Secrets)
+# 🔍 Debug print (optional - remove later)
 print("ENVIRONMENT VARIABLES:", list(os.environ.keys()))
 
-GROQ_API_KEY = os.environ.get("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")
+# ✅ Correct way to load secret
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
-MODEL_NAME = "llama-3.1-8b-instant"  # You can change this to a different supported model if needed
+MODEL_NAME = "llama-3.1-8b-instant"
 
-# Convert image to base64 (not sent, placeholder for future use or debugging)
+# Convert image to base64 (not actually used in API, placeholder)
 def image_to_base64(image: Image.Image) -> str:
     buffered = io.BytesIO()
     image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode()
 
-# Function to send defect info to Groq API and get analysis
+# Main function to analyze defect
 def analyze_defect(image, user_input):
     if image is None:
         return "⚠️ Please upload an image of the defective component."
 
     if not GROQ_API_KEY:
-        return "❌ GROQ_API_KEY not found. Please set it in Hugging Face 'Secrets'."
+        return "❌ Environment variable 'GROQ_API_KEY' is set in the system but empty."
 
     try:
+        # Optional: Keep for future image analysis
         image_b64 = image_to_base64(image)
 
         system_prompt = (
@@ -62,12 +64,12 @@ def analyze_defect(image, user_input):
             reply = result['choices'][0]['message']['content']
             return reply.strip()
         else:
-            return f"❌ Error: Groq API request failed with status code {response.status_code}."
+            return f"❌ Groq API request failed. Status code: {response.status_code}"
 
     except Exception as e:
-        return f"❌ An unexpected error occurred: {str(e)}"
+        return f"❌ An error occurred: {str(e)}"
 
-# Gradio UI definition
+# Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## 🛠️ NDT Defect Analysis Assistant (Powered by Groq AI)")
 
@@ -80,6 +82,5 @@ with gr.Blocks() as demo:
 
     analyze_button.click(fn=analyze_defect, inputs=[image_input, user_input], outputs=output)
 
-# Launch the app
 demo.launch()