Update app.py

app.py

@@ -1,4 +1,73 @@
-# ... (Previous code remains the same until the process_image function) ...
+import gradio as gr
+from PIL import Image
+import os
+from dotenv import load_dotenv
+from simple_salesforce import Salesforce
+from datetime import datetime
+import hashlib
+import shutil
+import base64
+import pytz
+import time
+
+# Load environment variables
+print("Loading environment variables at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
+load_dotenv()
+SF_USERNAME = os.getenv("SF_USERNAME")
+SF_PASSWORD = os.getenv("SF_PASSWORD")
+SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
+
+# Validate Salesforce credentials
+if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN]):
+    raise ValueError("Missing Salesforce credentials. Set SF_USERNAME, SF_PASSWORD, and SF_SECURITY_TOKEN in environment variables.")
+print("Salesforce credentials validated successfully.")
+
+# Initialize Salesforce connection
+try:
+    print("Attempting Salesforce connection at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
+    sf = Salesforce(
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN,
+        domain='login'
+    )
+    print("Salesforce connection established successfully.")
+except Exception as e:
+    print(f"Salesforce connection failed: {str(e)}")
+    raise
+
+print("Initializing application at", datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S IST"))
+
+# Updated Milestone Percentages
+milestone_percentage_map = {
+    "Excavation and Foundation": 10,
+    "Structural Framework": 40,
+    "Roofing": 70,
+    "Exterior Work": 85,
+    "Interior Work": 95,
+    "Final Completion": 100
+}
+
+# Adjust the timezone to your local timezone
+local_timezone = pytz.timezone("Asia/Kolkata")
+
+# AI model prediction (for simulation)
+def mock_ai_model(image):
+    img = image.convert("RGB")
+    max_size = 1024
+    img.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
+
+    img_bytes = img.tobytes()
+    img_hash = int(hashlib.sha256(img_bytes).hexdigest(), 16)
+
+    # Assign milestone based on hash (for simulation)
+    milestone_index = img_hash % len(milestone_percentage_map)
+    milestone = list(milestone_percentage_map.keys())[milestone_index]
+
+    confidence_raw = 0.85 + ((img_hash % 1000) / 1000) * (0.95 - 0.85)
+    confidence_score = round(confidence_raw, 2)
+
+    return milestone, milestone_percentage_map[milestone], confidence_score
 
 # Image processing and Salesforce upload for multiple images
 def process_image(images, project_name, project_type):
@@ -13,12 +82,11 @@ def process_image(images, project_name, project_type):
         image_urls = []
         milestone_confidences = []
         all_percentages = []
-        all_milestones = set()  # To store all unique milestones
+        all_milestones = set()
         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)
@@ -55,7 +123,7 @@ def process_image(images, project_name, project_type):
                     dominant_image_url = file_url
 
                 all_percentages.append(percent_complete)
-                all_milestones.add(milestone)  # Add milestone to set to avoid duplicates
+                all_milestones.add(milestone)
                 milestone_confidences.append((milestone, confidence_score))
                 results.append(f"Image {idx+1}: {milestone} - {percent_complete}% completion (Confidence: {confidence_score})")
 
@@ -67,7 +135,7 @@ def process_image(images, project_name, project_type):
             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
+        all_milestones_str = ", ".join(all_milestones)
 
         now = datetime.now(local_timezone)
         local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]
@@ -76,7 +144,7 @@ def process_image(images, project_name, project_type):
             "Name__c": project_name,
             "Project_Type__c": project_type,
             "Completion_Percentage__c": max_percent_complete,
-            "Current_Milestone__c": all_milestones_str,  # Store all milestones
+            "Current_Milestone__c": all_milestones_str,
             "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",
@@ -104,4 +172,84 @@ def process_image(images, project_name, project_type):
     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) ...
+# Gradio UI with enhanced styling
+with gr.Blocks(css="""
+    .gradio-container {
+        background-color: #f0f4f8;
+        font-family: Arial, sans-serif;
+    }
+    .title {
+        color: #2c3e50;
+        font-size: 24px;
+        text-align: center;
+        font-weight: bold;
+    }
+    .gradio-row {
+        text-align: center;
+    }
+    .gradio-container .output {
+        text-align: center;
+    }
+    .gradio-container .input {
+        text-align: center;
+    }
+    .gradio-container .button {
+        display: block;
+        margin: 0 auto;
+        background-color: #3498db;
+        color: white;
+        border: none;
+        padding: 10px 20px;
+        border-radius: 5px;
+        cursor: pointer;
+    }
+    .gradio-container .button:hover {
+        background-color: #2980b9;
+    }
+    progress::-webkit-progress-value {
+        background-color: #2ecc71;
+        border-radius: 5px;
+    }
+    progress::-webkit-progress-bar {
+        background-color: #ecf0f1;
+        border-radius: 5px;
+    }
+    details summary {
+        cursor: pointer;
+        padding: 10px;
+        background-color: #ecf0f1;
+        border-radius: 5px;
+    }
+    details ul {
+        margin-top: 10px;
+    }
+""") as demo:
+    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 (Required)", placeholder="e.g. Project_12345")
+        
+    image_input = gr.File(
+        file_count="multiple",
+        file_types=["image"],
+        label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB each)"
+    )
+    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)  # Block and enable debug mode
+    print("Gradio interface launched successfully.")
+    while True:  # Keep the script alive
+        time.sleep(10)
+except Exception as e:
+    print(f"Failed to launch Gradio interface: {str(e)}")
+    raise