Update app.py

app.py

@@ -41,7 +41,7 @@ local_timezone = pytz.timezone("Asia/Kolkata")
 def process_image(images, project_name):
     try:
         if not images or len(images) == 0:
-            return "Error: Please upload at least one image to proceed.", "Pending", "", 0
+            return "Error: Please upload at least one image to proceed.", "Pending", "", "", 0
 
         # Process each image
         milestones = []
@@ -49,9 +49,9 @@ def process_image(images, project_name):
             img = Image.open(image)
             image_size_mb = os.path.getsize(image) / (1024 * 1024)
             if image_size_mb > 20:
-                return "Error: One or more images exceed 20MB.", "Failure", "", 0
+                return "Error: One or more images exceed 20MB.", "Failure", "", "", 0
             if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
-                return "Error: Only JPG/PNG images are supported.", "Failure", "", 0
+                return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
             # Save image to public folder temporarily before uploading to Salesforce
             upload_dir = "public_uploads"
@@ -78,7 +78,7 @@ def process_image(images, project_name):
                 content_version_id = content_version_result['id']
                 file_url = f"https://sathkruthatechsolutionspri8-dev-ed.develop.lightning.force.com/{content_version_id}"
             except Exception as e:
-                return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", 0
+                return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
             # Mock AI model to detect milestone based on image content
             img_bytes = img.tobytes()
@@ -224,12 +224,12 @@ def process_image(images, project_name):
         try:
             sf.Construction__c.create(record)
         except Exception as e:
-            return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", 0
+            return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0
 
-        return result_html, "Success", final_milestone, f"{percent_complete}%"
+        return result_html, "Success", final_milestone, "", f"{percent_complete}%"
 
     except Exception as e:
-        return f"Error: {str(e)}", "Failure", "", "0%"
+        return f"Error: {str(e)}", "Failure", "", "", "0%"
 
 # Gradio UI with enhanced styling
 with gr.Blocks(css="""
@@ -292,12 +292,13 @@ with gr.Blocks(css="""
     output_html = gr.HTML(label="Result")  # Changed to HTML for richer output
     upload_status = gr.Textbox(label="Upload Status")
     milestone = gr.Textbox(label="Detected Milestone")
+    confidence = gr.Textbox(label="Confidence Score")  # Kept for compatibility, but unused
     progress = gr.Textbox(label="Completion Percentage", interactive=False)
 
     submit_button.click(
         fn=process_image,
         inputs=[image_input, project_name_input],
-        outputs=[output_html, upload_status, milestone, progress]
+        outputs=[output_html, upload_status, milestone, confidence, progress]
     )
 
 demo.launch(share=True)