Update app.py

app.py

@@ -1,43 +1,37 @@
 import os
 import gradio as gr
-import base64
 import requests
+import base64
 from PIL import Image
 from io import BytesIO
 
-# Set your Groq API Key
+# Set your Groq API key
 GROQ_API_KEY = os.getenv("GROQ_API_KEY") or "PASTE_YOUR_GROQ_API_KEY_HERE"
-GROQ_MODEL = "llama3-8b-8192"
 GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
+GROQ_MODEL = "llama3-70b-8192"
 
-history = []
-
-def analyze_image_with_prompt(image, user_prompt):
-    global history
-
+def analyze_damage(image, user_prompt):
     if image is None or not user_prompt.strip():
-        return "⚠️ Please upload an image and ask a question."
+        return "⚠️ Please upload an image and enter a question or description."
 
+    # Convert image to base64 for potential future use
     buffered = BytesIO()
     image.save(buffered, format="JPEG")
     img_base64 = base64.b64encode(buffered.getvalue()).decode()
 
-    if not history:
-        history.append({
-            "role": "system",
-            "content": (
-                "You are a professional construction assistant. Users upload images and describe damage. Based on the description, provide:\n"
-                "- Type of construction damage\n"
-                "- Possible causes\n"
-                "- Recommended tools/materials\n"
-                "- Estimated repair time"
-            )
-        })
+    system_prompt = (
+        "You are a helpful construction engineer. The user has uploaded an image of some construction damage "
+        "and has asked a question. Based on the image and their description or question, identify:\n"
+        "- Type of damage\n"
+        "- Possible causes\n"
+        "- Recommended repair materials/tools\n"
+        "- Estimated repair time"
+    )
 
-    history.append({
-        "role": "user",
-        "content": f"Image uploaded. Description: {user_prompt}"
-    })
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": f"Image uploaded.\nQuestion: {user_prompt}"}
+    ]
 
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
@@ -46,67 +40,53 @@ def analyze_image_with_prompt(image, user_prompt):
 
     payload = {
         "model": GROQ_MODEL,
-        "messages": history,
+        "messages": messages,
         "temperature": 0.7
     }
 
-    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}"
-
-def reset_session():
-    global history
-    history = []
-    return "✅ Session reset. Please upload a new image and ask a question."
-
-# UI Design
-with gr.Blocks(theme=gr.themes.Base(), css="""
-    #main-box {background-color: #f4f4f9; border-radius: 20px; padding: 2rem; box-shadow: 0 4px 12px rgba(0,0,0,0.1);}
-    h1, h2, h3 {color: #2c3e50;}
-    .gr-button {border-radius: 10px !important; font-weight: bold; font-size: 1rem;}
-    .gr-textbox textarea {font-size: 1rem;}
-    .gr-textbox label {font-weight: bold; color: #34495e;}
-    .gr-image label {font-weight: bold; color: #34495e;}
-""") as demo:
-
-    with gr.Column(elem_id="main-box"):
-        gr.Markdown("""
-        # 🏗️ Construction Damage Assistant
-        Upload a picture and describe the issue to get detailed construction repair guidance.
-        """)
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                img_input = gr.Image(
-                    type="pil",
-                    label="📤 Upload Picture of Damage",
-                    height=300
-                )
-
-                prompt_input = gr.Textbox(
-                    lines=4,
-                    label="📝 Ask Your Question or Describe the Problem",
-                    placeholder="e.g., What caused this crack? How to fix it?"
-                )
-
-                with gr.Row():
-                    submit_btn = gr.Button("🔍 Analyze Issue", variant="primary")
-                    reset_btn = gr.Button("🔁 Start New Session")
-
-            with gr.Column(scale=1):
-                output = gr.Textbox(
-                    label="🧠 AI Assistant's Response",
-                    lines=20,
-                    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])
-
-# Launch the app
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+# Gradio UI
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange")) as demo:
+    gr.Markdown("# 🏗️ Construction Damage Assistant")
+
+    gr.Markdown("""
+    Upload an image of the damage and ask a question like:<br>
+    🔹 *What type of crack is this?*<br>
+    ��� *What material should I use to fix it?*<br><br>
+    The assistant will analyze your question and give expert suggestions.
+    """, elem_classes="description-box")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            image_input = gr.Image(label="📸 Upload Damage Photo", type="pil", height=280)
+
+            question_input = gr.Textbox(
+                label="🧾 Ask a Question or Describe the Problem",
+                placeholder="e.g., What caused this crack? How do I fix it?",
+                lines=3
+            )
+
+            submit_btn = gr.Button("🔍 Analyze", variant="primary")
+            clear_btn = gr.Button("🧹 Clear Inputs")
+
+        with gr.Column(scale=1.5):
+            output = gr.Textbox(
+                label="💡 AI Response",
+                lines=18,
+                show_copy_button=True
+            )
+
+    submit_btn.click(fn=analyze_damage, inputs=[image_input, question_input], outputs=output)
+    clear_btn.click(lambda: (None, "", ""), outputs=[image_input, question_input, output])
+
+    gr.Markdown("___")
+    gr.Markdown("🔧 Powered by **Groq LLaMA3** | Built with ❤️ using **Gradio**")
+
 demo.launch()