Update app.py

app.py

@@ -5,12 +5,11 @@ import requests
 from PIL import Image
 from io import BytesIO
 
-# 🔐 Groq API key setup
+# 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"
 
-# 💬 Session history
 history = []
 
 def analyze_image_with_prompt(image, user_prompt):
@@ -19,27 +18,22 @@ def analyze_image_with_prompt(image, user_prompt):
     if image is None or not user_prompt.strip():
         return "⚠️ Please upload an image and ask a question."
 
-    # Encode image to base64 (placeholder, not used directly)
     buffered = BytesIO()
     image.save(buffered, format="JPEG")
     img_base64 = base64.b64encode(buffered.getvalue()).decode()
 
-    # System role prompt (set only once)
     if not history:
         history.append({
             "role": "system",
             "content": (
-                "You are a professional construction assistant.\n"
-                "Users upload images and describe issues.\n"
-                "Based on their description, provide:\n"
-                "- The type of construction damage\n"
+                "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"
             )
         })
 
-    # Add user message with image context
     history.append({
         "role": "user",
         "content": f"Image uploaded. Description: {user_prompt}"
@@ -63,43 +57,56 @@ def analyze_image_with_prompt(image, user_prompt):
         history.append({"role": "assistant", "content": reply})
         return reply
     else:
-        return f"❌ Error: {response.status_code}\n{response.text}"
+        return f"\u274c Error: {response.status_code}\n{response.text}"
 
 def reset_session():
     global history
     history = []
-    return "✅ Session reset. You can upload a new image and ask a question."
-
-# 🎨 Gradio App
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🏗️ Construction Assistant: Upload Image & Ask a Question")
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            img_input = gr.Image(
-                type="pil",
-                label="📤 Upload a Picture of the Problem"
-            )
-
-            prompt_input = gr.Textbox(
-                lines=3,
-                label="📝 Describe the Problem or Ask a Question",
-                placeholder="e.g. What caused this crack? How can I repair it?"
-            )
-
-            with gr.Row():
-                submit_btn = gr.Button("🔍 Get Analysis", variant="primary")
-                reset_btn = gr.Button("🔁 Start Over")
-
-        with gr.Column(scale=1):
-            output = gr.Textbox(
-                label="🧠 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
+    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
 demo.launch()