Update app.py

app.py

@@ -9,7 +9,7 @@ from io import BytesIO
 GROQ_API_KEY = os.getenv("GROQ_API_KEY") or "PASTE_YOUR_GROQ_API_KEY_HERE"
 
 # ✅ Groq LLaMA 3 model endpoint (text-only)
-GROQ_MODEL = "llama-3.3-70b-versatile"
+GROQ_MODEL = "llama-3-70b-8192"
 GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
 
 # Store history per session
@@ -18,28 +18,33 @@ history = []
 def analyze_image_with_prompt(image, user_prompt):
     global history
 
+    if image is None:
+        return "⚠️ Please upload an image."
+
+    if not user_prompt.strip():
+        return "⚠️ Please describe the damage or ask a question."
+
     # Convert image to base64 (for reference only, model won't see it)
     buffered = BytesIO()
     image.save(buffered, format="JPEG")
     img_base64 = base64.b64encode(buffered.getvalue()).decode()
 
-    # Add system message on first call
+    # Add system message once
     if len(history) == 0:
         history.append({
             "role": "system",
             "content": (
                 "You are a helpful construction engineer bot.\n"
-                "Users may upload an image, but since you can't see it,\n"
-                "ask them to describe the damage. Once described, identify:\n"
+                "The user has uploaded an image (you can't see it).\n"
+                "Ask them to describe the visible damage and respond with:\n"
                 "- Type of damage\n"
                 "- Possible causes\n"
                 "- Recommended tools/materials\n"
-                "- Estimated repair time.\n"
-                "Then answer follow-up questions."
+                "- Estimated repair time."
             )
         })
 
-    # Append user input (with note that image was uploaded)
+    # Add user message
     history.append({
         "role": "user",
         "content": f"(Image uploaded by user)\nPrompt: {user_prompt}"
@@ -53,62 +58,66 @@ def analyze_image_with_prompt(image, user_prompt):
     payload = {
         "model": GROQ_MODEL,
         "messages": history,
-        "temperature": 0.7
+        "temperature": 0.5
     }
 
-    response = requests.post(GROQ_URL, headers=headers, json=payload)
-
-    if response.status_code == 200:
+    try:
+        response = requests.post(GROQ_URL, headers=headers, json=payload)
+        response.raise_for_status()
         reply = response.json()["choices"][0]["message"]["content"]
         history.append({"role": "assistant", "content": reply})
         return reply
-    else:
-        return f"\u274c Error: {response.status_code}\n{response.text}"
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
 
 def reset_session():
     global history
     history = []
-    return "Session reset. Upload a new image and ask a question."
+    return "🔄 Session reset. Please upload a new image and describe the damage."
 
 # 🌐 Gradio Interface
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-        # 🏗️ Construction Damage Assistant
-        Upload an image and describe the damage. The AI will help you identify the issue and recommend repairs.
-    """)
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            img_input = gr.Image(
-                type="pil",
-                label="Upload Damage Image",
-                height=300
-            )
-
-            prompt_input = gr.Textbox(
-                lines=3,
-                label="Describe the Problem or Ask a Question",
-                placeholder="e.g. What type of crack is this? What caused it?"
-            )
-
-            with gr.Row():
-                submit_btn = gr.Button("🔍 Analyze", variant="primary")
-                reset_btn = gr.Button("🔁 New Session")
-
-        with gr.Column(scale=2):
-            output = gr.Textbox(
-                label="🧠 AI Assistant Response",
-                lines=18,
-                show_copy_button=True
-            )
-
-    submit_btn.click(analyze_image_with_prompt, inputs=[img_input, prompt_input], outputs=[output])
-    reset_btn.click(reset_session, outputs=[output])
-
-    gr.Markdown("""
-        <hr>
-        ⚠️ **Note**: The AI cannot directly analyze images. Please describe what you see.
-    """)
+def build_ui():
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+        gr.Markdown("""
+            # 🏗️ Construction Damage Assistant
+            Upload an image of damage and briefly describe it. AI will help you analyze and recommend fixes.
+        """)
+
+        with gr.Row():
+            with gr.Column(scale=1):
+                img_input = gr.Image(
+                    type="pil",
+                    label="📷 Upload Damage Image",
+                    height=300
+                )
+
+                prompt_input = gr.Textbox(
+                    lines=3,
+                    label="✏️ Describe the Damage or Ask a Question",
+                    placeholder="e.g. What type of crack is this? What caused it?"
+                )
+
+                output = gr.Textbox(
+                    label="🧠 AI Assistant Response",
+                    lines=18,
+                    show_copy_button=True
+                )
+
+                img_input.change(analyze_image_with_prompt, inputs=[img_input, prompt_input], outputs=output)
+                prompt_input.submit(analyze_image_with_prompt, inputs=[img_input, prompt_input], outputs=output)
+
+                with gr.Row():
+                    reset_btn = gr.Button("🔁 Reset Session")
+                reset_btn.click(reset_session, outputs=output)
+
+        gr.Markdown("""
+            <hr>
+            ⚠️ <small>The AI cannot see the image — your description is essential for accurate help.</small>
+        """)
+
+    return demo
 
 # 🚀 Launch the app
-demo.launch()
+if __name__ == "__main__":
+    ui = build_ui()
+    ui.launch()