Update app.py

app.py

@@ -5,14 +5,14 @@ import requests
 from PIL import Image
 from io import BytesIO
 
-# 🔐 Groq API Key (Set via environment variable or paste manually for local testing)
+# 🔐 Groq API Key
 GROQ_API_KEY = os.getenv("GROQ_API_KEY") or "PASTE_YOUR_GROQ_API_KEY_HERE"
 
-# ✅ Groq model info
+# ✅ Model details
 GROQ_MODEL = "llama-3.3-70b-versatile"
 GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
 
-# 🧠 Chat session memory
+# 🧠 Chat history
 history = []
 
 def analyze_image_with_prompt(image, user_prompt):
@@ -21,18 +21,13 @@ def analyze_image_with_prompt(image, user_prompt):
     if not image or not user_prompt.strip():
         return "⚠️ Please upload an image and enter a question or prompt."
 
-    # Convert image to base64
-    buffered = BytesIO()
-    image.save(buffered, format="JPEG")
-    img_base64 = base64.b64encode(buffered.getvalue()).decode()
-
-    # Add system prompt if first message
+    # 💡 Only include user prompt — not base64 image
     if len(history) == 0:
         system_msg = {
             "role": "system",
             "content": (
                 "You are a helpful construction engineer bot. "
-                "Given an image of a construction issue and a prompt, identify:\n"
+                "Given a prompt describing a construction issue (e.g., crack, damage), identify:\n"
                 "- Type of damage\n"
                 "- Possible causes\n"
                 "- Recommended tools and materials\n"
@@ -41,13 +36,14 @@ def analyze_image_with_prompt(image, user_prompt):
             )
         }
         history.append(system_msg)
-        user_msg = {
-            "role": "user",
-            "content": f"Image base64: {img_base64}\nPrompt: {user_prompt}"
-        }
-        history.append(user_msg)
-    else:
-        history.append({"role": "user", "content": user_prompt})
+
+    # Add user message
+    history.append({"role": "user", "content": user_prompt})
+
+    # ✅ Limit chat history to avoid hitting token limits
+    MAX_TURNS = 5  # 5 user-assistant exchanges
+    if len(history) > (MAX_TURNS * 2 + 1):  # +1 for system
+        history = [history[0]] + history[-(MAX_TURNS * 2):]
 
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
@@ -78,15 +74,15 @@ def reset_history():
 with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container { max-width: 950px; margin: auto; }") as demo:
     gr.Markdown("## 🏗️ Construction Damage Identifier Chatbot")
     gr.Markdown(
-        "Upload a photo of construction damage (e.g., cracks, corrosion) and ask a question. "
-        "The AI will try to diagnose the problem and suggest solutions."
+        "Upload a photo of construction damage (for reference) and ask a question. "
+        "The AI will try to diagnose the issue based on your prompt."
     )
 
     with gr.Row():
         with gr.Column(scale=1):
             img_input = gr.Image(
                 type="pil",
-                label="🖼️ Upload Image of Damage",
+                label="🖼️ Upload Image (for your reference)",
                 height=300
             )
             prompt_input = gr.Textbox(
@@ -110,7 +106,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container { max-width: 950px
     submit_btn.click(analyze_image_with_prompt, inputs=[img_input, prompt_input], outputs=[output])
     reset_btn.click(reset_history, outputs=[output, prompt_input])
 
-    gr.Markdown("➡️ You can continue chatting with the AI about the same image.")
+    gr.Markdown("➡️ You can continue chatting with the AI about the same issue.")
 
-# 🚀 Launch app
 demo.launch()