Update app.py

app.py

@@ -68,7 +68,7 @@ def analyze_image_with_hf(image_path):
 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.", None, "Pending", "", "", 0
 
         results_html = []
         upload_statuses = []
@@ -81,14 +81,14 @@ def process_image(images, project_name):
                 img = Image.open(image)
             except Exception as e:
                 if "cannot identify image file" in str(e):
-                    return f"Error: Unsupported image format '{os.path.basename(image)}'. Please use JPG, JPEG, PNG, or convert to a supported format.", "Failure", "", "", 0
-                return f"Error: Failed to open image '{os.path.basename(image)}' - {str(e)}", "Failure", "", "", 0
+                    return f"Error: Unsupported image format '{os.path.basename(image)}'. Please use JPG, JPEG, PNG, or convert to a supported format.", None, "Failure", "", "", 0
+                return f"Error: Failed to open image '{os.path.basename(image)}' - {str(e)}", None, "Failure", "", "", 0
 
             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.", None, "Failure", "", "", 0
             if not str(image).lower().endswith(('.jpg', '.jpeg', '.png', '.avif')):
-                return "Error: Only JPG, JPEG, PNG, or AVIF images are supported.", "Failure", "", "", 0
+                return "Error: Only JPG, JPEG, PNG, or AVIF images are supported.", None, "Failure", "", "", 0
 
             upload_dir = "public_uploads"
             os.makedirs(upload_dir, exist_ok=True)
@@ -114,7 +114,7 @@ def process_image(images, project_name):
                 if "STORAGE_LIMIT_EXCEEDED" in str(e):
                     file_url = "Upload skipped due to storage limit"
                 else:
-                    return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
+                    return f"Error: Failed to upload image to Salesforce - {str(e)}", None, "Failure", "", "", 0
 
             detected_elements = analyze_image_with_hf(image)
 
@@ -184,14 +184,10 @@ def process_image(images, project_name):
             completed_html = "".join([f'<li style="color: green;">✔ {task}</li>' for task in completed_tasks])
             not_completed_html = "".join([f'<li style="color: red;">✘ {task}</li>' for task in not_completed_tasks])
 
-            # Add image preview to the HTML output
-            image_preview_html = f'<img src="file://{saved_image_path}" style="max-width: 100%; height: auto; margin-top: 10px;" alt="Uploaded Image">'
-
             if final_milestone == "Completed":
                 result_html = f"""
                 <div style="font-family: Arial, sans-serif; padding: 20px; background-color: #f9f9f9; border-radius: 10px; margin-bottom: 20px;">
                     <h2 style="color: #2c3e50; text-align: center;">Project Summary - {os.path.basename(image)}</h2>
-                    {image_preview_html}
                     <div style="display: flex; justify-content: space-around; margin-bottom: 20px;">
                         <div style="text-align: center;">
                             <h3 style="color: #34495e;">Detected Milestone</h3>
@@ -217,14 +213,13 @@ def process_image(images, project_name):
                             {completed_html}
                         </ul>
                     </details>
-                    <p style="color: green;">Project is fully completed as of 02:16 PM IST, June 20, 2025.</p>
+                    <p style="color: green;">Project is fully completed as of 02:22 PM IST, June 20, 2025.</p>
                 </div>
                 """
             else:
                 result_html = f"""
                 <div style="font-family: Arial, sans-serif; padding: 20px; background-color: #f9f9f9; border-radius: 10px; margin-bottom: 20px;">
                     <h2 style="color: #2c3e50; text-align: center;">Project Summary - {os.path.basename(image)}</h2>
-                    {image_preview_html}
                     <div style="display: flex; justify-content: space-around; margin-bottom: 20px;">
                         <div style="text-align: center;">
                             <h3 style="color: #34495e;">Detected Milestone</h3>
@@ -256,7 +251,7 @@ def process_image(images, project_name):
                             {not_completed_html}
                         </ul>
                     </details>
-                    <p style="color: orange;">Construction is in progress at {final_milestone} stage as of 02:16 PM IST, June 20, 2025.</p>
+                    <p style="color: orange;">Construction is in progress at {final_milestone} stage as of 02:22 PM IST, June 20, 2025.</p>
                 </div>
                 """
 
@@ -267,8 +262,8 @@ def process_image(images, project_name):
 
         combined_html = "<div>" + "".join(results_html) + "</div>"
 
-        # Set current time to 02:16 PM IST, June 20, 2025
-        now = local_timezone.localize(datetime(2025, 6, 20, 14, 16))
+        # Set current time to 02:22 PM IST, June 20, 2025
+        now = local_timezone.localize(datetime(2025, 6, 20, 14, 22))
         local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]
 
         record = {
@@ -285,14 +280,14 @@ def process_image(images, project_name):
             if file_url and file_url != "Upload skipped due to storage limit":
                 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)}", None, "Failure", "", "", 0
 
-        return combined_html, ",".join(upload_statuses), ",".join(milestones), "", ",".join(progresses)
+        return combined_html, Image.open(saved_image_path) if images else None, ",".join(upload_statuses), ",".join(milestones), "", ",".join(progresses)
 
     except Exception as e:
-        return f"Error: {str(e)}", "Failure", "", "", "0%"
+        return f"Error: {str(e)}", None, "Failure", "", "", "0%"
 
-# Gradio UI with enhanced styling
+# Gradio UI with enhanced styling and image preview
 with gr.Blocks(css="""
     .gradio-container {
         background-color: #f0f4f8;
@@ -351,6 +346,7 @@ with gr.Blocks(css="""
 
     submit_button = gr.Button("Process Image")
     output_html = gr.HTML(label="Result")
+    output_image = gr.Image(label="Uploaded Image Preview")  # Added image preview component
     upload_status = gr.Textbox(label="Upload Status")
     milestone = gr.Textbox(label="Detected Milestone")
     confidence = gr.Textbox(label="Confidence Score")  # Kept for compatibility, but unused
@@ -359,7 +355,7 @@ with gr.Blocks(css="""
     submit_button.click(
         fn=process_image,
         inputs=[image_input, project_name_input],
-        outputs=[output_html, upload_status, milestone, confidence, progress]
+        outputs=[output_html, output_image, upload_status, milestone, confidence, progress]
     )
 
 demo.launch(share=True)