Create app.py

app.py

@@ -0,0 +1,95 @@
+import gradio as gr
+import requests
+from PIL import Image
+from io import BytesIO
+import os
+
+# ✅ Load API key from environment
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+GROQ_MODEL = "llama3-70b-8192"
+
+# 🔧 Simulated caption (replace later)
+def get_caption(image):
+    return "This image shows a large crack in the wall extending diagonally."
+
+# 💬 Send prompt to Groq
+def generate_response(image, user_query=""):
+    if GROQ_API_KEY is None:
+        return "❌ API key not set. Please set GROQ_API_KEY using os.environ."
+
+    caption = get_caption(image)
+
+    prompt = f"""
+You are a construction damage analysis assistant. Based on the description: "{caption}", and user question: "{user_query}", provide the following:
+1. Type of damage
+2. Likely cause
+3. Suggested repair solutions
+4. Required tools
+5. Estimated time and cost
+
+Respond in a helpful tone.
+"""
+
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}",
+        "Content-Type": "application/json"
+    }
+
+    data = {
+        "model": GROQ_MODEL,
+        "messages": [{"role": "user", "content": prompt}]
+    }
+
+    response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=data)
+    
+    if response.status_code == 200:
+        return response.json()['choices'][0]['message']['content']
+    else:
+        return f"❌ Error {response.status_code}: {response.text}"
+
+# ============================
+# 🌟 Gradio Modern UI
+# ============================
+
+with gr.Blocks(css="""
+.gradio-container {
+  background-color: #f8f9fa;
+  font-family: 'Segoe UI', sans-serif;
+}
+#title {
+  text-align: center;
+  font-size: 32px;
+  font-weight: bold;
+  padding: 20px 0;
+  color: #333;
+}
+#subtitle {
+  text-align: center;
+  font-size: 16px;
+  color: #666;
+  margin-bottom: 20px;
+}
+.card {
+  border: 1px solid #ddd;
+  border-radius: 16px;
+  background-color: white;
+  padding: 20px;
+  box-shadow: 0 4px 8px rgba(0,0,0,0.03);
+}
+""") as demo:
+
+    gr.Markdown("<div id='title'>🏗️ BuildFix AI — Construction Damage Inspector</div>")
+    gr.Markdown("<div id='subtitle'>Upload a construction damage image and get expert analysis with repair suggestions instantly.</div>")
+
+    with gr.Row():
+        with gr.Column(scale=1, elem_classes="card"):
+            image_input = gr.Image(type="pil", label="📸 Upload Image of Damage")
+            user_query = gr.Textbox(label="💬 Ask a follow-up question (optional)", placeholder="e.g., How expensive is this repair?", lines=2)
+            submit_btn = gr.Button("🔍 Analyze", variant="primary")
+
+        with gr.Column(scale=1, elem_classes="card"):
+            result_output = gr.Textbox(label="🧠 AI Report", lines=20)
+
+    submit_btn.click(fn=generate_response, inputs=[image_input, user_query], outputs=result_output)
+
+demo.launch()