Spaces:
Sleeping
Sleeping
| 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 | |
| # 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") | |
| # 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 Salesforce connection | |
| try: | |
| sf = Salesforce( | |
| username=SF_USERNAME, | |
| password=SF_PASSWORD, | |
| security_token=SF_SECURITY_TOKEN, | |
| domain='login' | |
| ) | |
| except Exception as e: | |
| 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 | |
| # Mock AI model to detect milestone based on image content | |
| img_bytes = img.tobytes() | |
| img_hash = int(hashlib.sha256(img_bytes).hexdigest(), 16) | |
| milestone_index = img_hash % len(VALID_MILESTONES) | |
| milestone = VALID_MILESTONES[milestone_index] | |
| # Adjust milestone detection based on internal/external features | |
| if milestone == "Walls Erected" and any("interior" in img_text.lower() for img_text in [os.path.basename(image)] if img_text): | |
| if any(keyword in str(os.path.basename(image)).lower() for keyword in ["electrical", "plumbing", "wiring", "pipes"]): | |
| milestone = "Completed" | |
| else: | |
| milestone = "Walls Erected" | |
| 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: | |
| sf.Construction__c.create(record) | |
| except Exception as e: | |
| 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) | |