Update app.py

app.py

@@ -110,26 +110,42 @@ def process_image(image, project_name):
         # Detailed Completion Breakdown (based on detected milestone)
         completion_details = {
             "Planning": {
-                "completed": "Project outline and goals set, initial designs done.",
-                "not_completed": "Detailed plans, permits, and contractor hiring pending."
+                "completed": [
+                    "Project outline and goals set, initial designs done."
+                ],
+                "not_completed": [
+                    "Detailed plans, permits, and contractor hiring pending."
+                ]
             },
             "Foundation": {
-                "completed": "Foundation work is complete, concrete is poured.",
-                "not_completed": "Plumbing, electrical groundwork not yet done."
+                "completed": [
+                    "Foundation work is complete, concrete is poured."
+                ],
+                "not_completed": [
+                    "Plumbing, electrical groundwork not yet done."
+                ]
             },
             "Walls Erected": {
-                "completed": "All structural walls are up.",
-                "not_completed": "Roofing, windows, and internal walls are not yet finished."
+                "completed": [
+                    "All structural walls are up."
+                ],
+                "not_completed": [
+                    "Roofing, windows, and internal walls are not yet finished."
+                ]
             },
             "Completed": {
-                "completed": "All phases of the project are finished, including final touches.",
-                "not_completed": "There should be no more pending work."
+                "completed": [
+                    "All phases of the project are finished, including final touches."
+                ],
+                "not_completed": [
+                    "There should be no more pending work."
+                ]
             }
         }
 
         # Get the detailed completion information based on the detected milestone
-        completed_work = completion_details[milestone]["completed"]
-        not_completed_work = completion_details[milestone]["not_completed"]
+        completed_work = "\n".join([f"{idx+1}. {task}" for idx, task in enumerate(completion_details[milestone]["completed"])])
+        not_completed_work = "\n".join([f"{idx+1}. {task}" for idx, task in enumerate(completion_details[milestone]["not_completed"])])
 
         # Adjust the current time to local timezone
         local_time = datetime.now(local_timezone).strftime("%Y-%m-%dT%H:%M:%S%z")  # Corrected ISO 8601 format with timezone
@@ -151,15 +167,14 @@ def process_image(image, project_name):
         except Exception as e:
             return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0
 
-        # Return the detailed result
+        # Return the detailed result in the desired format
         result = f"""
-        Milestone: {milestone}
-        Completion Percentage: {percent_complete}%
+        Completed:
         
-        **Completed Work**:
         {completed_work}
 
-        **Not Completed Work**:
+        Not Completed:
+
         {not_completed_work}
 
         Confidence Score: {confidence_score * 100}%