Update app.py

app.py

@@ -1,99 +1,107 @@
-import gradio as gr
-from PIL import Image
-import time
-from datetime import datetime
-import pytz
-
-print("Starting application at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-
-# Simplified mock AI model
-def mock_ai_model(image):
-    print("Processing image in mock_ai_model at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-    img = image.convert("RGB")
-    # Basic heuristic: assume interior work for now (adjustable)
-    milestone = "Interior Work"
-    percent_complete = 95
-    confidence_score = 0.92
-    print(f"Image processed: {milestone}, {percent_complete}%, Confidence: {confidence_score}")
-    return milestone, percent_complete, confidence_score
-
-# Image processing function with debugging
+# ... (Previous code remains the same until the process_image function) ...
+
+# Image processing and Salesforce upload for multiple images
 def process_image(images, project_name, project_type):
-    print("Starting process_image at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-    if not images:
-        print("No images uploaded")
-        return "<p style='color: red;'>No images uploaded.</p>", "Pending", "", "", "0%"
-    if not project_name:
-        print("No project name provided")
-        return "<p style='color: red;'>Project Name is required.</p>", "Pending", "", "", "0%"
-
-    results = []
-    all_percentages = []
-    all_milestones = set()
-    max_percent_complete = 0
-    dominant_milestone = None
-
-    for idx, image_path in enumerate(images):
+    try:
+        if not images:
+            return "<p style='color: red;'>Error: Please upload at least one image to proceed.</p>", "Pending", "", "", 0
+
+        if not project_name:
+            return "<p style='color: red;'>Error: Project Name is required.</p>", "Pending", "", "", 0
+
+        results = []
+        image_urls = []
+        milestone_confidences = []
+        all_percentages = []
+        all_milestones = set()  # To store all unique milestones
+        max_percent_complete = 0
+        dominant_milestone = None
+        dominant_image_url = None
+
+        # Process each image
+        for idx, image_path in enumerate(images):
+            try:
+                img = Image.open(image_path)
+                milestone, percent_complete, confidence_score = mock_ai_model(img)
+
+                upload_dir = "public_uploads"
+                os.makedirs(upload_dir, exist_ok=True)
+                unique_id = datetime.now().strftime("%Y%m%d%H%M%S")
+                image_filename = f"{unique_id}_{idx}_{os.path.basename(image_path)}"
+                saved_image_path = os.path.join(upload_dir, image_filename)
+                shutil.copy(image_path, saved_image_path)
+
+                with open(saved_image_path, 'rb') as image_file:
+                    image_data = base64.b64encode(image_file.read()).decode('utf-8')
+
+                content_version = {
+                    'Title': image_filename,
+                    'PathOnClient': saved_image_path,
+                    'VersionData': image_data
+                }
+
+                try:
+                    content_version_result = sf.ContentVersion.create(content_version)
+                    content_version_id = content_version_result['id']
+                    file_url = f"https://sathkruthatechsolutionspri8-dev-ed.develop.lightning.force.com/{content_version_id}"
+                    image_urls.append(file_url)
+                except Exception as e:
+                    results.append(f"Image {idx+1}: Failed to upload to Salesforce - {str(e)}")
+                    continue
+
+                if percent_complete > max_percent_complete:
+                    max_percent_complete = percent_complete
+                    dominant_milestone = milestone
+                    dominant_image_url = file_url
+
+                all_percentages.append(percent_complete)
+                all_milestones.add(milestone)  # Add milestone to set to avoid duplicates
+                milestone_confidences.append((milestone, confidence_score))
+                results.append(f"Image {idx+1}: {milestone} - {percent_complete}% completion (Confidence: {confidence_score})")
+
+            except Exception as e:
+                results.append(f"Image {idx+1}: Error processing image - {str(e)}")
+                continue
+
+        if not results:
+            return "<p style='color: red;'>Error: No images were successfully processed.</p>", "Failure", "", "", 0
+
+        average_percent_complete = round(sum(all_percentages) / len(all_percentages), 2) if all_percentages else 0
+        all_milestones_str = ", ".join(all_milestones)  # Convert set to comma-separated string
+
+        now = datetime.now(local_timezone)
+        local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]
+
+        record = {
+            "Name__c": project_name,
+            "Project_Type__c": project_type,
+            "Completion_Percentage__c": max_percent_complete,
+            "Current_Milestone__c": all_milestones_str,  # Store all milestones
+            "Last_Updated_On__c": local_time,
+            "Upload_Status__c": "Success",
+            "Comments__c": f"Project {project_name} at {dominant_milestone or 'no milestone detected'} with {max_percent_complete}% completion",
+            "Last_Updated_Image__c": dominant_image_url or ""
+        }
+
         try:
-            print(f"Processing image {idx+1} at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-            img = Image.open(image_path)
-            milestone, percent_complete, confidence_score = mock_ai_model(img)
-            results.append(f"Image {idx+1}: {milestone} - {percent_complete}% completion (Confidence: {confidence_score})")
-            all_percentages.append(percent_complete)
-            all_milestones.add(milestone)
-            if percent_complete > max_percent_complete:
-                max_percent_complete = percent_complete
-                dominant_milestone = milestone
+            sf.Construction__c.create(record)
         except Exception as e:
-            print(f"Error processing image {idx+1}: {str(e)}")
-            results.append(f"Image {idx+1}: Error - {str(e)}")
-
-    average_percent_complete = round(sum(all_percentages) / len(all_percentages), 2) if all_percentages else 0
-    all_milestones_str = ", ".join(all_milestones)
-
-    output_html = "<div style='text-align: left;'>"
-    output_html += "<h3>Processing Results:</h3><ul>"
-    for result in results:
-        output_html += f"<li>{result}</li>"
-    output_html += "</ul>"
-    output_html += f"<p><strong>Project Record Created:</strong> {project_name} - {dominant_milestone or 'No milestone'} (Highest: {max_percent_complete}%, Average: {average_percent_complete}%, Milestones: {all_milestones_str})</p>"
-    output_html += "</div>"
-
-    # Fixed confidence score logic
-    confidence_scores = [0.90] if not all_percentages else [0.90 for _ in all_milestones]  # Default to 0.90
-    max_confidence = str(max(confidence_scores))
-
-    print("Finished process_image at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-    return output_html, "Success", dominant_milestone or "", max_confidence, f"{max_percent_complete}%"
-
-# Gradio UI
-with gr.Blocks(css="""
-    .gradio-container { background-color: #f0f4f8; font-family: Arial; }
-    .title { color: #2c3e50; font-size: 24px; text-align: center; font-weight: bold; }
-""") as demo:
-    print("Setting up Gradio interface at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-    gr.Markdown("<h1 class='title'>Construction Progress Analyzer</h1>")
-    
-    with gr.Row():
-        project_type_input = gr.Dropdown(label="Project Type", choices=["House", "Apartment"], value="House")
-        project_name_input = gr.Textbox(label="Project Name", placeholder="e.g. Project_12345")
-        
-    image_input = gr.File(file_count="multiple", file_types=["image"], label="Upload Images")
-    submit_button = gr.Button("Process Images")
-    output_html = gr.HTML(label="Result")
-    
-    submit_button.click(
-        fn=process_image,
-        inputs=[image_input, project_name_input, project_type_input],
-        outputs=[output_html]
-    )
-
-print("Launching Gradio interface at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-try:
-    demo.launch(share=True, debug=True)
-    print("Gradio interface launched successfully at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
-    while True:
-        time.sleep(10)
-except Exception as e:
-    print(f"Failed to launch Gradio interface: {str(e)}")
-    raise
+            return f"<p style='color: red;'>Error: Failed to update Salesforce - {str(e)}</p>", "Failure", "", "", 0
+
+        output_html = "<div style='text-align: left;'>"
+        output_html += "<h3>Processing Results:</h3><ul>"
+        for result in results:
+            output_html += f"<li>{result}</li>"
+        output_html += "</ul>"
+        output_html += f"<p><strong>Project Record Created:</strong> {project_name} - {dominant_milestone or 'No milestone detected'} (Highest: {max_percent_complete}%, Average: {average_percent_complete}%, Milestones: {all_milestones_str})</p>"
+        output_html += f"<p><strong>Note:</strong> Only the image with the highest completion percentage is stored in Salesforce due to field length constraints.</p>"
+        if not dominant_milestone:
+            output_html += "<p><strong>Warning:</strong> No valid milestone was set as dominant due to restricted picklist constraints (prior to conversion).</p>"
+        output_html += "</div>"
+
+        return output_html, "Success", dominant_milestone or "", f"Max Confidence: {max([conf for _, conf in milestone_confidences], default=0)}", f"{max_percent_complete}%"
+
+    except Exception as e:
+        return f"<p style='color: red;'>Error: {str(e)}</p>", "Failure", "", "", "0%"
+
+# ... (Rest of the code, including Gradio UI, remains the same) ...