Update app.py

app.py

@@ -50,7 +50,7 @@ def analyze_construction_image(image):
                 "content": [
                     {
                         "type": "text",
-                        "text": "Analyze this construction site image. Identify any safety issues or hazards, categorize them, provide a detailed description, and suggest steps to resolve them. Format your response as a JSON object with a key 'issues' containing an array of issue objects. Each issue object should have 'category', 'description', and 'resolution' keys."
+                        "text": "Analyze this construction site image. Identify any safety issues or hazards, categorize them, provide a detailed description, and suggest steps to resolve them. Format your response as a structured analysis with clear categories, descriptions, and resolution steps."
                     },
                     {
                         "type": "image_url",
@@ -70,7 +70,6 @@ def analyze_construction_image(image):
             max_tokens=1000,
             top_p=1,
             stream=False,
-            response_format={"type": "json_object"},
             stop=None
         )
 
@@ -78,23 +77,21 @@ def analyze_construction_image(image):
         result = completion.choices[0].message.content
         logger.debug(f"Raw API response: {result}")
 
-        # Parse the result as JSON
-        try:
-            parsed_result = json.loads(result)
-            issues = parsed_result.get('issues', [])
-        except json.JSONDecodeError:
-            logger.error("Failed to parse API response as JSON")
-            return [(None, "Error: Invalid response format")]
-
-        # Format the analysis results
+        # Parse the result
         analysis_text = "Image Analysis Results:\n\n"
-        for issue in issues:
-            analysis_text += f"Category: {issue.get('category', 'N/A')}\n"
-            analysis_text += f"Description: {issue.get('description', 'N/A')}\n"
-            analysis_text += "Resolution Steps:\n"
-            for step in issue.get('resolution', ['N/A']):
-                analysis_text += f"- {step}\n"
-            analysis_text += "\n"
+        categories = result.split("Category: ")[1:]  # Split by categories
+        for category in categories:
+            lines = category.split("\n")
+            category_name = lines[0].strip()
+            analysis_text += f"Category: {category_name}\n"
+            
+            description_start = category.find("Description: ") + len("Description: ")
+            description_end = category.find("Resolution Steps:")
+            description = category[description_start:description_end].strip()
+            analysis_text += f"Description: {description}\n"
+            
+            resolution_steps = "".join(lines[lines.index("Resolution Steps:") + 1:]).replace("* ", "").replace("\n", "")
+            analysis_text += f"Resolution Steps:\n{resolution_steps}\n\n"
 
         logger.info("Analysis completed successfully")
         return [(None, analysis_text)]