Update app.py

app.py

@@ -26,16 +26,12 @@ def analyze_construction_image(image):
         completion = client.chat.completions.create(
             model="llama-3.2-11b-vision-preview",
             messages=[
-                {
-                    "role": "system",
-                    "content": "You are an AI assistant specialized in analyzing construction site images. Identify issues, categorize them, and provide steps to resolve them."
-                },
                 {
                     "role": "user",
                     "content": [
                         {
                             "type": "text",
-                            "text": "Analyze this construction image. Identify the snag category, provide a detailed snag description, and list steps to desnag."
+                            "text": "You are an AI assistant specialized in analyzing construction site images. Analyze this construction image. Identify the snag category, provide a detailed snag description, and list steps to desnag. Format your response as follows:\nSnag Category: [category]\nSnag Description: [detailed description]\nSteps to Desnag:\n1. [step 1]\n2. [step 2]\n3. [step 3]"
                         },
                         {
                             "type": "image_url",
@@ -47,7 +43,7 @@ def analyze_construction_image(image):
                 }
             ],
             temperature=0.7,
-            max_tokens=300,
+            max_tokens=500,
             top_p=1,
             stream=False,
             stop=None
@@ -56,10 +52,17 @@ def analyze_construction_image(image):
         result = completion.choices[0].message.content
 
         # Parse the result
-        lines = result.split('\n')
-        snag_category = lines[0] if len(lines) > 0 else "N/A"
-        snag_description = lines[1] if len(lines) > 1 else "N/A"
-        desnag_steps = "\n".join(lines[2:]) if len(lines) > 2 else "N/A"
+        snag_category = "N/A"
+        snag_description = "N/A"
+        desnag_steps = "N/A"
+
+        for line in result.split('\n'):
+            if line.startswith("Snag Category:"):
+                snag_category = line.split(":", 1)[1].strip()
+            elif line.startswith("Snag Description:"):
+                snag_description = line.split(":", 1)[1].strip()
+            elif line.startswith("Steps to Desnag:"):
+                desnag_steps = "\n".join(result.split("Steps to Desnag:")[1].strip().split("\n"))
 
         return snag_category, snag_description, desnag_steps
     except Exception as e: