Update app.py

app.py

@@ -1,165 +1,296 @@
-import streamlit as st
-import requests
-import base64
-import json
-from datetime import datetime
-import boto3
+import gradio as gr
 from PIL import Image
-import io
 import os
 from dotenv import load_dotenv
 from simple_salesforce import Salesforce
+from datetime import datetime
+import hashlib
+import shutil
+import base64
+import pytz
 
 # Load environment variables
 load_dotenv()
+SF_USERNAME = os.getenv("SF_USERNAME")
+SF_PASSWORD = os.getenv("SF_PASSWORD")
+SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 
-# Configuration
-AI_MODEL_ENDPOINT = os.getenv("AI_MODEL_ENDPOINT", "https://your-huggingface-or-lambda-endpoint/predict-milestone")
-S3_BUCKET = os.getenv("S3_BUCKET", "your-s3-bucket")
-AWS_REGION = os.getenv("AWS_REGION", "us-east-1")
-MAX_FILE_SIZE = 20 * 1024 * 1024  # 20MB in bytes
-MIN_IMAGE_DIMENSIONS = (800, 600)  # Minimum width and height for clarity
+# 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.")
 
-# Initialize AWS S3 client
-try:
-    s3_client = boto3.client(
-        "s3",
-        region_name=AWS_REGION,
-        aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),
-        aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY")
-    )
-except Exception as e:
-    st.error(f"Failed to initialize S3 client: {str(e)}")
-    st.stop()
-
-# Initialize Salesforce client
+# Initialize Salesforce connection
 try:
     sf = Salesforce(
-        username=os.getenv("SF_USERNAME"),
-        password=os.getenv("SF_PASSWORD"),
-        security_token=os.getenv("SF_SECURITY_TOKEN"),
-        domain="login"  # Use "test" for sandbox
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN,
+        domain='login'
     )
 except Exception as e:
-    st.error(f"Failed to connect to Salesforce: {str(e)}")
-    st.stop()
-
-# Streamlit app
-st.title("Construction Progress Verifier")
-st.write("Upload one or more construction site photos to track project milestones.")
-
-# Input for Salesforce Construction__c Record ID
-project_id = st.text_input("Enter Salesforce Construction ID (Construction__c Record ID)", "")
-
-# File uploader for multiple images
-uploaded_files = st.file_uploader("Choose images (JPG/PNG, <20MB each)", type=["jpg", "png"], accept_multiple_files=True)
-
-if uploaded_files and project_id:
-    # Initialize results list for output
-    results = []
-    progress_bar = st.progress(0)
-    status_text = st.empty()
-    total_images = len(uploaded_files)
-    progress_increment = 100 // (total_images * 3)  # Divide progress into upload, AI, and Salesforce steps
-
-    for idx, uploaded_file in enumerate(uploaded_files, 1):
-        # Validate file size
-        if uploaded_file.size > MAX_FILE_SIZE:
-            results.append({
-                "filename": uploaded_file.name,
-                "status": "Failed",
-                "message": "File size exceeds 20MB. Please upload a smaller image."
-            })
-            continue
+    print(f"Salesforce connection failed: {str(e)}")
+    raise
+
+# Valid milestones
+VALID_MILESTONES = ["Planning", "Foundation", "Walls Erected", "Completed"]
+
+# Adjust the timezone to your local timezone
+local_timezone = pytz.timezone("Asia/Kolkata")
+
+# Image processing and Salesforce upload
+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
+
+        # Process each image
+        milestones = []
+        for image in images:
+            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
+            if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
+                return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
+
+            # Save image to public folder temporarily before uploading to Salesforce
+            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}_{os.path.basename(image)}"
+            saved_image_path = os.path.join(upload_dir, image_filename)
+            shutil.copy(image, saved_image_path)
+
+            # Convert image to base64 before uploading to Salesforce
+            with open(saved_image_path, 'rb') as image_file:
+                image_data = base64.b64encode(image_file.read()).decode('utf-8')
+
+            # Create the ContentVersion record in Salesforce
+            content_version = {
+                'Title': image_filename,
+                'PathOnClient': saved_image_path,
+                'VersionData': image_data
+            }
+
+            # Upload the file to Salesforce
+            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}"
+            except Exception as e:
+                return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
+
+            # Use file name to detect milestones as a fallback (simple keyword-based detection)
+            milestone = "Planning"  # Default if no match is found
+            image_filename_lower = image_filename.lower()
+
+            if any(keyword in image_filename_lower for keyword in ["foundation", "excavation", "digging"]):
+                milestone = "Foundation"
+            elif any(keyword in image_filename_lower for keyword in ["walls", "framing", "structure"]):
+                milestone = "Walls Erected"
+            elif any(keyword in image_filename_lower for keyword in ["completed", "final", "finished"]):
+                milestone = "Completed"
+
+            milestones.append(milestone)
+
+        # Determine overall milestone (most advanced detected)
+        final_milestone = max(set(milestones), key=milestones.count) if milestones else "Planning"
+        milestone_completion_map = {
+            "Planning": 10,
+            "Foundation": 30,
+            "Walls Erected": 50,
+            "Completed": 100,
+        }
+        percent_complete = milestone_completion_map.get(final_milestone, 0)
+
+        completion_details = {
+            "Planning": {
+                "completed": [
+                    "Initial project outline and objectives have been established.",
+                    "Preliminary designs and architectural plans are drafted.",
+                    "Stakeholder meetings and initial approvals are completed."
+                ],
+                "not_completed": [
+                    "Detailed construction plans and blueprints are pending finalization.",
+                    "Permits and regulatory approvals are yet to be obtained.",
+                    "Contractor selection and procurement processes are not yet complete."
+                ]
+            },
+            "Foundation": {
+                "completed": [
+                    "Site preparation, including clearing and leveling, is finished.",
+                    "Excavation for the foundation has been completed.",
+                    "Concrete pouring for the foundation, including footings and slabs, is done.",
+                    "Initial structural inspections for the foundation have been passed."
+                ],
+                "not_completed": [
+                    "Plumbing and electrical groundwork installations are pending.",
+                    "Backfilling and site grading around the foundation are not yet done.",
+                    "Above-ground structural work, such as columns and walls, has not started."
+                ]
+            },
+            "Walls Erected": {
+                "completed": [
+                    "The concrete framework, including columns and beams, is in place.",
+                    "All structural walls have been erected and stabilized.",
+                    "Temporary scaffolding and safety measures are installed for ongoing work.",
+                    "Initial inspections for structural integrity have been completed."
+                ],
+                "not_completed": [
+                    "Roofing installation and weatherproofing are pending.",
+                    "Windows, doors, and exterior cladding are not yet installed.",
+                    "Interior walls, electrical, and plumbing systems are still to be implemented."
+                ]
+            },
+            "Completed": {
+                "completed": [
+                    "The concrete framework, including columns, beams, and floor slabs, is fully constructed.",
+                    "Exterior walls, windows, and cladding are installed, completing the building's facade.",
+                    "Interior work, including electrical, plumbing, and HVAC systems, is fully implemented.",
+                    "Finishing touches, such as flooring, painting, and fixtures, are completed.",
+                    "All phases of the project are finished, including final inspections and approvals."
+                ],
+                "not_completed": [
+                    "There should be no more pending work as the project is fully completed."
+                ]
+            }
+        }
+
+        completed_tasks = completion_details[final_milestone]["completed"]
+        not_completed_tasks = completion_details[final_milestone]["not_completed"]
+
+        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])
+
+        result_html = f"""
+        <div style="font-family: Arial, sans-serif; padding: 20px; background-color: #f9f9f9; border-radius: 10px;">
+            <h2 style="color: #2c3e50; text-align: center;">Project Summary</h2>
+            <div style="display: flex; justify-content: space-around; margin-bottom: 20px;">
+                <div style="text-align: center;">
+                    <h3 style="color: #34495e;">Detected Milestone</h3>
+                    <p style="font-size: 18px; font-weight: bold;">{final_milestone}</p>
+                </div>
+                <div style="text-align: center;">
+                    <h3 style="color: #34495e;">Completion</h3>
+                    <progress value="{percent_complete}" max="100" style="width: 200px; height: 20px;"></progress>
+                    <p>{percent_complete}%</p>
+                </div>
+            </div>
+
+            <h3 style="color: #2c3e50;">Milestone Timeline</h3>
+            <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 20px;">
+                <span style="color: {'#2ecc71' if final_milestone == 'Planning' else '#bdc3c7'};">Planning</span>
+                <span style="color: {'#2ecc71' if final_milestone == 'Foundation' else '#bdc3c7'};">Foundation</span>
+                <span style="color: {'#2ecc71' if final_milestone == 'Walls Erected' else '#bdc3c7'};">Walls Erected</span>
+                <span style="color: {'#2ecc71' if final_milestone == 'Completed' else '#bdc3c7'};">Completed</span>
+            </div>
+
+            <details style="margin-bottom: 20px;">
+                <summary style="color: #2c3e50; font-weight: bold;">Completed Tasks</summary>
+                <ul style="padding-left: 20px;">
+                    {completed_html}
+                </ul>
+            </details>
+
+            <details style="margin-bottom: 20px;">
+                <summary style="color: #2c3e50; font-weight: bold;">Not Completed Tasks</summary>
+                <ul style="padding-left: 20px;">
+                    {not_completed_html}
+                </ul>
+            </details>
+        </div>
+        """
+
+        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,
+            "Current_Milestone__c": final_milestone,
+            "Completion_Percentage__c": percent_complete,
+            "Last_Updated_On__c": local_time,
+            "Upload_Status__c": "Success",
+            "Comments__c": f"{final_milestone}",
+            "Last_Updated_Image__c": file_url
+        }
 
         try:
-            # Read and validate image
-            status_text.text(f"Processing image {idx}/{total_images}: Validating {uploaded_file.name}...")
-            progress_bar.progress(min((idx - 1) * progress_increment * 3 + progress_increment, 100))
-            image = Image.open(uploaded_file)
-            if image.size < MIN_IMAGE_DIMENSIONS:
-                results.append({
-                    "filename": uploaded_file.name,
-                    "status": "Failed",
-                    "message": "Image resolution too low. Please upload a higher resolution image (min 800x600)."
-                })
-                continue
-
-            # Convert image to base64
-            buffered = io.BytesIO()
-            image.save(buffered, format="JPEG")
-            img_str = base64.b64encode(buffered.getvalue()).decode()
-
-            # Upload image to S3
-            status_text.text(f"Processing image {idx}/{total_images}: Uploading {uploaded_file.name} to storage...")
-            s3_key = f"construction_images/{datetime.now().strftime('%Y%m%d_%H%M%S')}_{uploaded_file.name}"
-            s3_client.put_object(Bucket=S3_BUCKET, Key=s3_key, Body=buffered.getvalue())
-            s3_url = f"https://{S3_BUCKET}.s3.{AWS_REGION}.amazonaws.com/{s3_key}"
-            progress_bar.progress(min((idx - 1) * progress_increment * 3 + 2 * progress_increment, 100))
-
-            # Call AI model endpoint
-            status_text.text(f"Processing image {idx}/{total_images}: Analyzing {uploaded_file.name} with AI model...")
-            payload = {"image": img_str}
-            ai_response = requests.post(
-                AI_MODEL_ENDPOINT,
-                json=payload,
-                headers={"Authorization": f"Bearer {os.getenv('AI_API_KEY')}"},
-                timeout=5
-            )
-
-            if ai_response.status_code == 200:
-                ai_result = ai_response.json()
-                milestone = ai_result.get("milestone", "Unknown")
-
-                # Update Salesforce record using simple-salesforce
-                status_text.text(f"Processing image {idx}/{total_images}: Updating Salesforce for {uploaded_file.name}...")
-                sf.Construction__c.update(project_id, {
-                    "Current_Milestone__c": milestone,
-                    "Last_Updated_Image__c": s3_url,
-                    "Upload_Status__c": "Success"
-                })
-
-                results.append({
-                    "filename": uploaded_file.name,
-                    "status": "Success",
-                    "milestone": milestone,
-                    "s3_url": s3_url,
-                    "message": "Salesforce record updated successfully."
-                })
-                progress_bar.progress(min(idx * progress_increment * 3, 100))
-            else:
-                results.append({
-                    "filename": uploaded_file.name,
-                    "status": "Failed",
-                    "message": f"AI model error: {ai_response.text}. Please retake the photo or try again."
-                })
-                progress_bar.progress(min(idx * progress_increment * 3, 100))
+            sf.Construction__c.create(record)
         except Exception as e:
-            results.append({
-                "filename": uploaded_file.name,
-                "status": "Failed",
-                "message": f"Error processing image: {str(e)}. Please retake the photo or contact support."
-            })
-            progress_bar.progress(min(idx * progress_increment * 3, 100))
-
-    # Display results
-    st.markdown("### Processing Results")
-    for result in results:
-        if result["status"] == "Success":
-            st.success(
-                f"**{result['filename']}**: "
-                f"Milestone: {result['milestone']}, "
-                f"S3 URL: {result['s3_url']}, "
-                f"Message: {result['message']}"
-            )
-        else:
-            st.error(f"**{result['filename']}**: {result['message']}")
-
-elif uploaded_files and not project_id:
-    st.warning("Please enter a valid Salesforce Construction ID.")
-elif project_id and not uploaded_files:
-    st.warning("Please upload at least one image.")
-
-# Footer
-st.markdown("---")
-st.write("Developed by Sathkrutha Tech Solutions | © 2025")
+            return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0
+
+        return result_html, "Success", final_milestone, f"{percent_complete}%"
+
+    except Exception as e:
+        return f"Error: {str(e)}", "Failure", "", "", "0%"
+
+# Gradio UI
+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():
+        image_input = gr.Files(type="filepath", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)")
+        project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")
+
+    submit_button = gr.Button("Process Image")
+    output_html = gr.HTML(label="Result")
+    upload_status = gr.Textbox(label="Upload Status")
+    milestone = gr.Textbox(label="Detected Milestone")
+    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]
+    )
+
+demo.launch(share=True)