Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import google.generativeai as genai | |
| import cv2 | |
| import numpy as np | |
| from PIL import Image, ImageDraw, ImageFont | |
| import os | |
| from dotenv import load_dotenv | |
| import requests | |
| import base64 | |
| import io | |
| import json | |
| import time | |
| import logging | |
| from typing import Optional, Tuple, List | |
| import warnings | |
| warnings.filterwarnings("ignore") | |
| # Configure logging | |
| logging.basicConfig(level=logging.INFO) | |
| logger = logging.getLogger(__name__) | |
| # Load environment variables | |
| load_dotenv() | |
| # Configure Gemini API | |
| try: | |
| genai.configure(api_key=os.getenv("GEMINI_API_KEY")) | |
| logger.info("Gemini API configured successfully") | |
| except Exception as e: | |
| logger.warning(f"Gemini API configuration failed: {e}") | |
| # Configure ElevenLabs API | |
| try: | |
| from elevenlabs import generate, set_api_key | |
| if os.getenv("ELEVENLABS_API_KEY"): | |
| set_api_key(os.getenv("ELEVENLABS_API_KEY")) | |
| logger.info("ElevenLabs API configured successfully") | |
| except ImportError: | |
| logger.warning("ElevenLabs not available - voice features disabled") | |
| except Exception as e: | |
| logger.warning(f"ElevenLabs API configuration failed: {e}") | |
| class BuildTheFuture: | |
| def __init__(self): | |
| self.model = None | |
| self.yolo_model = None | |
| self._initialize_models() | |
| def _initialize_models(self): | |
| """Initialize AI models with error handling""" | |
| try: | |
| self.model = genai.GenerativeModel('gemini-2.0-flash-exp') | |
| logger.info("Gemini model initialized successfully") | |
| except Exception as e: | |
| logger.error(f"Failed to initialize Gemini model: {e}") | |
| self.model = None | |
| def load_yolo_model(self): | |
| """Load YOLOv11 model for structural detection""" | |
| try: | |
| from ultralytics import YOLO | |
| # You can replace this with your custom trained model | |
| self.yolo_model = YOLO('yolov11n.pt') | |
| return True | |
| except Exception as e: | |
| print(f"Error loading YOLO model: {e}") | |
| return False | |
| def detect_structures(self, image): | |
| """Detect structural elements using YOLOv11""" | |
| if self.yolo_model is None: | |
| if not self.load_yolo_model(): | |
| return image, "YOLO model not available" | |
| try: | |
| # Convert PIL to OpenCV format | |
| img_cv = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) | |
| # Run YOLO detection | |
| results = self.yolo_model(img_cv) | |
| # Draw bounding boxes | |
| annotated_img = results[0].plot() | |
| # Convert back to PIL | |
| annotated_pil = Image.fromarray(cv2.cvtColor(annotated_img, cv2.COLOR_BGR2RGB)) | |
| return annotated_pil, "Structures detected successfully" | |
| except Exception as e: | |
| return image, f"Detection error: {str(e)}" | |
| def complete_construction(self, image, style="realistic"): | |
| """Complete the construction using Gemini 2.5 Flash Image""" | |
| if self.model is None: | |
| return image, "Gemini model not available. Please check your API key." | |
| try: | |
| # Prepare the prompt based on style | |
| style_prompts = { | |
| "realistic": "Complete this unfinished construction site realistically. Fill in the missing parts with appropriate building materials, proper architectural details, and realistic finishing touches. Make it look like a completed, functional building. Ensure the completion looks natural and follows proper construction practices.", | |
| "futuristic": "Transform this unfinished construction site into a futuristic, high-tech building. Add modern architectural elements, glass facades, smart building features, solar panels, and futuristic design elements. Make it look like a building from the year 2050 with advanced technology integration.", | |
| "artistic": "Complete this construction site with artistic and creative architectural elements. Add unique design features, creative materials, colorful elements, and artistic touches that make it visually striking and memorable. Think outside the box with creative architecture." | |
| } | |
| prompt = style_prompts.get(style, style_prompts["realistic"]) | |
| # Add additional context for better results | |
| enhanced_prompt = f"{prompt} The image should maintain the same perspective and lighting as the original. Focus on completing the unfinished parts while maintaining architectural coherence." | |
| # Generate the completed image | |
| response = self.model.generate_content([enhanced_prompt, image]) | |
| # Extract the generated image | |
| if response.candidates and response.candidates[0].content.parts: | |
| for part in response.candidates[0].content.parts: | |
| if hasattr(part, 'inline_data') and part.inline_data: | |
| # Convert base64 to image | |
| image_data = base64.b64decode(part.inline_data.data) | |
| completed_image = Image.open(io.BytesIO(image_data)) | |
| return completed_image, f"Construction completed successfully with {style} style" | |
| return image, "No image generated - please try again" | |
| except Exception as e: | |
| logger.error(f"Construction completion error: {e}") | |
| return image, f"Completion error: {str(e)}" | |
| def generate_voice_narration(self, style): | |
| """Generate voice narration using ElevenLabs (optional)""" | |
| if not os.getenv("ELEVENLABS_API_KEY"): | |
| logger.info("ElevenLabs API key not found - voice narration disabled") | |
| return None | |
| try: | |
| narration_texts = { | |
| "realistic": "Here's how your construction project will look when completed with realistic finishing touches and proper architectural details. The AI has filled in the missing parts with appropriate materials and construction techniques.", | |
| "futuristic": "Behold the future! Your construction site has been transformed into a cutting-edge, high-tech building of tomorrow. Notice the modern architectural elements and smart building features.", | |
| "artistic": "Watch as your construction site becomes a masterpiece of creative architecture and artistic design. The AI has added unique design features and creative elements that make it visually striking." | |
| } | |
| text = narration_texts.get(style, narration_texts["realistic"]) | |
| audio = generate(text=text, voice="Rachel", model="eleven_monolingual_v1") | |
| return audio | |
| except Exception as e: | |
| logger.error(f"Voice generation error: {e}") | |
| return None | |
| def create_comparison_image(self, original, completed): | |
| """Create a side-by-side comparison image""" | |
| if original is None or completed is None: | |
| return None | |
| try: | |
| # Resize images to same height | |
| height = min(original.height, completed.height) | |
| width = min(original.width, completed.width) | |
| original_resized = original.resize((width, height), Image.Resampling.LANCZOS) | |
| completed_resized = completed.resize((width, height), Image.Resampling.LANCZOS) | |
| # Create comparison image | |
| comparison_width = width * 2 + 20 # Add gap between images | |
| comparison_height = height + 60 # Add space for labels | |
| comparison_img = Image.new('RGB', (comparison_width, comparison_height), 'white') | |
| # Paste images | |
| comparison_img.paste(original_resized, (0, 30)) | |
| comparison_img.paste(completed_resized, (width + 20, 30)) | |
| # Add labels | |
| draw = ImageDraw.Draw(comparison_img) | |
| try: | |
| font = ImageFont.truetype("arial.ttf", 20) | |
| except: | |
| font = ImageFont.load_default() | |
| draw.text((width//2 - 50, 5), "BEFORE", fill='black', font=font) | |
| draw.text((width + 20 + width//2 - 50, 5), "AFTER", fill='black', font=font) | |
| return comparison_img | |
| except Exception as e: | |
| logger.error(f"Comparison image creation error: {e}") | |
| return None | |
| # Initialize the app | |
| app = BuildTheFuture() | |
| def process_image(uploaded_image, completion_style): | |
| """Main processing function for the Gradio interface""" | |
| if uploaded_image is None: | |
| return None, None, None, None, "Please upload an image first" | |
| try: | |
| # Step 1: Detect structures | |
| detected_image, detection_msg = app.detect_structures(uploaded_image) | |
| # Step 2: Complete construction | |
| completed_image, completion_msg = app.complete_construction(uploaded_image, completion_style) | |
| # Step 3: Create comparison image | |
| comparison_image = app.create_comparison_image(uploaded_image, completed_image) | |
| # Step 4: Generate voice narration | |
| audio = app.generate_voice_narration(completion_style) | |
| status_msg = f"✅ Detection: {detection_msg}\n✅ Completion: {completion_msg}" | |
| if audio is None: | |
| status_msg += "\n💡 Voice narration is optional (ElevenLabs API key not configured)" | |
| return uploaded_image, detected_image, completed_image, comparison_image, status_msg, audio | |
| except Exception as e: | |
| logger.error(f"Processing error: {e}") | |
| return uploaded_image, uploaded_image, uploaded_image, None, f"❌ Error: {str(e)}", None | |
| # Create Gradio interface | |
| with gr.Blocks(title="BuildTheFuture: AI-Powered Construction Completion", theme=gr.themes.Soft()) as demo: | |
| gr.Markdown(""" | |
| # 🏗️ BuildTheFuture: AI-Powered Completion of Unfinished Constructions | |
| Upload a photo of an unfinished construction site and watch AI complete it with realistic, futuristic, or artistic finishes! | |
| **How it works:** | |
| 1. Upload an image of an unfinished building, road, or bridge | |
| 2. Select your preferred completion style | |
| 3. View the original, detected structures, and AI-completed result | |
| 4. Use the comparison slider to see before/after | |
| """) | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| image_input = gr.Image( | |
| label="Upload Unfinished Construction Photo", | |
| type="pil", | |
| height=300 | |
| ) | |
| style_selector = gr.Radio( | |
| choices=["realistic", "futuristic", "artistic"], | |
| value="realistic", | |
| label="Completion Style", | |
| info="Choose how you want the construction to be completed" | |
| ) | |
| process_btn = gr.Button("🚀 Complete Construction", variant="primary", size="lg") | |
| status_text = gr.Textbox( | |
| label="Status", | |
| interactive=False, | |
| lines=3 | |
| ) | |
| with gr.Column(scale=2): | |
| with gr.Tabs(): | |
| with gr.Tab("Original"): | |
| original_output = gr.Image(label="Original Unfinished Site", height=400) | |
| with gr.Tab("Detected Structures"): | |
| detected_output = gr.Image(label="YOLO Detection Overlay", height=400) | |
| with gr.Tab("AI Completed"): | |
| completed_output = gr.Image(label="AI-Completed Construction", height=400) | |
| with gr.Tab("Before vs After"): | |
| gr.Markdown("### Interactive Comparison") | |
| comparison = gr.Image( | |
| label="Drag slider to compare before and after", | |
| height=400, | |
| show_download_button=True | |
| ) | |
| # Voice narration section | |
| with gr.Row(): | |
| audio_output = gr.Audio( | |
| label="Voice Narration", | |
| visible=True | |
| ) | |
| # Add examples with all sample images | |
| with gr.Row(): | |
| gr.Markdown("### 📸 Try These Sample Construction Images") | |
| gr.Examples( | |
| examples=[ | |
| # Renamed construction samples | |
| ["samples/building_001.jpg", "realistic"], | |
| ["samples/building_001.jpg", "futuristic"], | |
| ["samples/building_001.jpg", "artistic"], | |
| ["samples/building_002.jpg", "realistic"], | |
| ["samples/building_002.jpg", "futuristic"], | |
| ["samples/building_002.jpg", "artistic"], | |
| ["samples/building_003.jpg", "realistic"], | |
| ["samples/building_003.jpg", "futuristic"], | |
| ["samples/building_003.jpg", "artistic"], | |
| ["samples/residential_004.jpg", "realistic"], | |
| ["samples/residential_004.jpg", "futuristic"], | |
| ["samples/residential_004.jpg", "artistic"], | |
| ["samples/residential_005.jpg", "realistic"], | |
| ["samples/residential_005.jpg", "futuristic"], | |
| ["samples/residential_005.jpg", "artistic"], | |
| ["samples/residential_006.jpg", "realistic"], | |
| ["samples/residential_006.jpg", "futuristic"], | |
| ["samples/residential_006.jpg", "artistic"], | |
| ["samples/residential_007.jpg", "realistic"], | |
| ["samples/residential_007.jpg", "futuristic"], | |
| ["samples/residential_007.jpg", "artistic"], | |
| ["samples/residential_008.jpg", "realistic"], | |
| ["samples/residential_008.jpg", "futuristic"], | |
| ["samples/residential_008.jpg", "artistic"], | |
| ["samples/residential_009.jpg", "realistic"], | |
| ["samples/residential_009.jpg", "futuristic"], | |
| ["samples/residential_009.jpg", "artistic"], | |
| ["samples/commercial_010.jpg", "realistic"], | |
| ["samples/commercial_010.jpg", "futuristic"], | |
| ["samples/commercial_010.jpg", "artistic"], | |
| ["samples/commercial_011.jpg", "realistic"], | |
| ["samples/commercial_011.jpg", "futuristic"], | |
| ["samples/commercial_011.jpg", "artistic"], | |
| ["samples/commercial_012.jpg", "realistic"], | |
| ["samples/commercial_012.jpg", "futuristic"], | |
| ["samples/commercial_012.jpg", "artistic"], | |
| ["samples/infrastructure_013.jpg", "realistic"], | |
| ["samples/infrastructure_013.jpg", "futuristic"], | |
| ["samples/infrastructure_013.jpg", "artistic"], | |
| ["samples/infrastructure_014.jpg", "realistic"], | |
| ["samples/infrastructure_014.jpg", "futuristic"], | |
| ["samples/infrastructure_014.jpg", "artistic"], | |
| ["samples/infrastructure_015.jpg", "realistic"], | |
| ["samples/infrastructure_015.jpg", "futuristic"], | |
| ["samples/infrastructure_015.jpg", "artistic"], | |
| ["samples/infrastructure_016.jpg", "realistic"], | |
| ["samples/infrastructure_016.jpg", "futuristic"], | |
| ["samples/infrastructure_016.jpg", "artistic"], | |
| ["samples/infrastructure_017.jpg", "realistic"], | |
| ["samples/infrastructure_017.jpg", "futuristic"], | |
| ["samples/infrastructure_017.jpg", "artistic"], | |
| ["samples/infrastructure_018.jpg", "realistic"], | |
| ["samples/infrastructure_018.jpg", "futuristic"], | |
| ["samples/infrastructure_018.jpg", "artistic"], | |
| ["samples/construction_019.jpg", "realistic"], | |
| ["samples/construction_019.jpg", "futuristic"], | |
| ["samples/construction_019.jpg", "artistic"], | |
| ["samples/construction_020.jpg", "realistic"], | |
| ["samples/construction_020.jpg", "futuristic"], | |
| ["samples/construction_020.jpg", "artistic"], | |
| ["samples/construction_021.jpg", "realistic"], | |
| ["samples/construction_021.jpg", "futuristic"], | |
| ["samples/construction_021.jpg", "artistic"], | |
| ["samples/construction_022.jpg", "realistic"], | |
| ["samples/construction_022.jpg", "futuristic"], | |
| ["samples/construction_022.jpg", "artistic"], | |
| ["samples/construction_023.jpg", "realistic"], | |
| ["samples/construction_023.jpg", "futuristic"], | |
| ["samples/construction_023.jpg", "artistic"], | |
| ["samples/construction_024.jpg", "realistic"], | |
| ["samples/construction_024.jpg", "futuristic"], | |
| ["samples/construction_024.jpg", "artistic"], | |
| ["samples/building_025.jpg", "realistic"], | |
| ["samples/building_025.jpg", "futuristic"], | |
| ["samples/building_025.jpg", "artistic"], | |
| ["samples/building_026.jpg", "realistic"], | |
| ["samples/building_026.jpg", "futuristic"], | |
| ["samples/building_026.jpg", "artistic"], | |
| ["samples/building_027.jpg", "realistic"], | |
| ["samples/building_027.jpg", "futuristic"], | |
| ["samples/building_027.jpg", "artistic"], | |
| ["samples/construction_028.jpg", "realistic"], | |
| ["samples/construction_028.jpg", "futuristic"], | |
| ["samples/construction_028.jpg", "artistic"], | |
| ["samples/construction_029.jpg", "realistic"], | |
| ["samples/construction_029.jpg", "futuristic"], | |
| ["samples/construction_029.jpg", "artistic"], | |
| ["samples/construction_030.jpg", "realistic"], | |
| ["samples/construction_030.jpg", "futuristic"], | |
| ["samples/construction_030.jpg", "artistic"], | |
| ["samples/construction_031.jpg", "realistic"], | |
| ["samples/construction_031.jpg", "futuristic"], | |
| ["samples/construction_031.jpg", "artistic"], | |
| ["samples/construction_032.jpg", "realistic"], | |
| ["samples/construction_032.jpg", "futuristic"], | |
| ["samples/construction_032.jpg", "artistic"], | |
| ["samples/construction_033.jpg", "realistic"], | |
| ["samples/construction_033.jpg", "futuristic"], | |
| ["samples/construction_033.jpg", "artistic"], | |
| ["samples/construction_034.jpg", "realistic"], | |
| ["samples/construction_034.jpg", "futuristic"], | |
| ["samples/construction_034.jpg", "artistic"], | |
| ["samples/construction_035.jpg", "realistic"], | |
| ["samples/construction_035.jpg", "futuristic"], | |
| ["samples/construction_035.jpg", "artistic"], | |
| ["samples/construction_036.jpg", "realistic"], | |
| ["samples/construction_036.jpg", "futuristic"], | |
| ["samples/construction_036.jpg", "artistic"], | |
| ["samples/construction_037.jpg", "realistic"], | |
| ["samples/construction_037.jpg", "futuristic"], | |
| ["samples/construction_037.jpg", "artistic"], | |
| ["samples/construction_038.jpg", "realistic"], | |
| ["samples/construction_038.jpg", "futuristic"], | |
| ["samples/construction_038.jpg", "artistic"], | |
| ["samples/construction_039.jpg", "realistic"], | |
| ["samples/construction_039.jpg", "futuristic"], | |
| ["samples/construction_039.jpg", "artistic"], | |
| ["samples/construction_040.jpg", "realistic"], | |
| ["samples/construction_040.jpg", "futuristic"], | |
| ["samples/construction_040.jpg", "artistic"], | |
| ["samples/construction_041.jpg", "realistic"], | |
| ["samples/construction_041.jpg", "futuristic"], | |
| ["samples/construction_041.jpg", "artistic"], | |
| ["samples/construction_042.jpg", "realistic"], | |
| ["samples/construction_042.jpg", "futuristic"], | |
| ["samples/construction_042.jpg", "artistic"], | |
| ["samples/construction_043.jpg", "realistic"], | |
| ["samples/construction_043.jpg", "futuristic"], | |
| ["samples/construction_043.jpg", "artistic"], | |
| ["samples/construction_044.jpg", "realistic"], | |
| ["samples/construction_044.jpg", "futuristic"], | |
| ["samples/construction_044.jpg", "artistic"], | |
| ["samples/construction_045.jpg", "realistic"], | |
| ["samples/construction_045.jpg", "futuristic"], | |
| ["samples/construction_045.jpg", "artistic"], | |
| ["samples/construction_046.jpg", "realistic"], | |
| ["samples/construction_046.jpg", "futuristic"], | |
| ["samples/construction_046.jpg", "artistic"], | |
| ["samples/construction_047.jpg", "realistic"], | |
| ["samples/construction_047.jpg", "futuristic"], | |
| ["samples/construction_047.jpg", "artistic"], | |
| ["samples/construction_048.jpg", "realistic"], | |
| ["samples/construction_048.jpg", "futuristic"], | |
| ["samples/construction_048.jpg", "artistic"], | |
| ["samples/bridge_049.jpg", "realistic"], | |
| ["samples/bridge_049.jpg", "futuristic"], | |
| ["samples/bridge_049.jpg", "artistic"], | |
| ["samples/bridge_050.jpg", "realistic"], | |
| ["samples/bridge_050.jpg", "futuristic"], | |
| ["samples/bridge_050.jpg", "artistic"], | |
| ["samples/bridge_051.jpg", "realistic"], | |
| ["samples/bridge_051.jpg", "futuristic"], | |
| ["samples/bridge_051.jpg", "artistic"], | |
| ["samples/construction_052.jpg", "realistic"], | |
| ["samples/construction_052.jpg", "futuristic"], | |
| ["samples/construction_052.jpg", "artistic"], | |
| ["samples/construction_053.jpg", "realistic"], | |
| ["samples/construction_053.jpg", "futuristic"], | |
| ["samples/construction_053.jpg", "artistic"], | |
| ["samples/construction_054.jpg", "realistic"], | |
| ["samples/construction_054.jpg", "futuristic"], | |
| ["samples/construction_054.jpg", "artistic"], | |
| ["samples/construction_055.jpg", "realistic"], | |
| ["samples/construction_055.jpg", "futuristic"], | |
| ["samples/construction_055.jpg", "artistic"], | |
| ["samples/construction_056.jpg", "realistic"], | |
| ["samples/construction_056.jpg", "futuristic"], | |
| ["samples/construction_056.jpg", "artistic"], | |
| ["samples/construction_057.jpg", "realistic"], | |
| ["samples/construction_057.jpg", "futuristic"], | |
| ["samples/construction_057.jpg", "artistic"], | |
| ["samples/construction_058.jpg", "realistic"], | |
| ["samples/construction_058.jpg", "futuristic"], | |
| ["samples/construction_058.jpg", "artistic"], | |
| ["samples/construction_059.jpg", "realistic"], | |
| ["samples/construction_059.jpg", "futuristic"], | |
| ["samples/construction_059.jpg", "artistic"], | |
| ["samples/construction_060.jpg", "realistic"], | |
| ["samples/construction_060.jpg", "futuristic"], | |
| ["samples/construction_060.jpg", "artistic"], | |
| ["samples/construction_061.jpg", "realistic"], | |
| ["samples/construction_061.jpg", "futuristic"], | |
| ["samples/construction_061.jpg", "artistic"], | |
| ["samples/construction_062.jpg", "realistic"], | |
| ["samples/construction_062.jpg", "futuristic"], | |
| ["samples/construction_062.jpg", "artistic"], | |
| ["samples/construction_063.jpg", "realistic"], | |
| ["samples/construction_063.jpg", "futuristic"], | |
| ["samples/construction_063.jpg", "artistic"], | |
| ["samples/construction_064.jpg", "realistic"], | |
| ["samples/construction_064.jpg", "futuristic"], | |
| ["samples/construction_064.jpg", "artistic"], | |
| ["samples/construction_065.jpg", "realistic"], | |
| ["samples/construction_065.jpg", "futuristic"], | |
| ["samples/construction_065.jpg", "artistic"], | |
| ["samples/construction_066.jpg", "realistic"], | |
| ["samples/construction_066.jpg", "futuristic"], | |
| ["samples/construction_066.jpg", "artistic"], | |
| ["samples/construction_067.jpg", "realistic"], | |
| ["samples/construction_067.jpg", "futuristic"], | |
| ["samples/construction_067.jpg", "artistic"], | |
| ["samples/construction_068.jpg", "realistic"], | |
| ["samples/construction_068.jpg", "futuristic"], | |
| ["samples/construction_068.jpg", "artistic"], | |
| ["samples/construction_069.jpg", "realistic"], | |
| ["samples/construction_069.jpg", "futuristic"], | |
| ["samples/construction_069.jpg", "artistic"], | |
| ["samples/construction_070.jpg", "realistic"], | |
| ["samples/construction_070.jpg", "futuristic"], | |
| ["samples/construction_070.jpg", "artistic"], | |
| ["samples/construction_071.jpg", "realistic"], | |
| ["samples/construction_071.jpg", "futuristic"], | |
| ["samples/construction_071.jpg", "artistic"], | |
| ["samples/construction_072.jpg", "realistic"], | |
| ["samples/construction_072.jpg", "futuristic"], | |
| ["samples/construction_072.jpg", "artistic"], | |
| ["samples/construction_073.jpg", "realistic"], | |
| ["samples/construction_073.jpg", "futuristic"], | |
| ["samples/construction_073.jpg", "artistic"], | |
| ["samples/construction_074.jpg", "realistic"], | |
| ["samples/construction_074.jpg", "futuristic"], | |
| ["samples/construction_074.jpg", "artistic"], | |
| ["samples/construction_075.jpg", "realistic"], | |
| ["samples/construction_075.jpg", "futuristic"], | |
| ["samples/construction_075.jpg", "artistic"], | |
| ["samples/construction_076.jpg", "realistic"], | |
| ["samples/construction_076.jpg", "futuristic"], | |
| ["samples/construction_076.jpg", "artistic"], | |
| ["samples/construction_077.jpg", "realistic"], | |
| ["samples/construction_077.jpg", "futuristic"], | |
| ["samples/construction_077.jpg", "artistic"], | |
| ["samples/construction_078.jpg", "realistic"], | |
| ["samples/construction_078.jpg", "futuristic"], | |
| ["samples/construction_078.jpg", "artistic"], | |
| ["samples/construction_079.jpg", "realistic"], | |
| ["samples/construction_079.jpg", "futuristic"], | |
| ["samples/construction_079.jpg", "artistic"], | |
| ["samples/construction_080.jpg", "realistic"], | |
| ["samples/construction_080.jpg", "futuristic"], | |
| ["samples/construction_080.jpg", "artistic"], | |
| ["samples/construction_081.jpg", "realistic"], | |
| ["samples/construction_081.jpg", "futuristic"], | |
| ["samples/construction_081.jpg", "artistic"], | |
| ["samples/construction_082.jpg", "realistic"], | |
| ["samples/construction_082.jpg", "futuristic"], | |
| ["samples/construction_082.jpg", "artistic"], | |
| ["samples/construction_083.jpg", "realistic"], | |
| ["samples/construction_083.jpg", "futuristic"], | |
| ["samples/construction_083.jpg", "artistic"], | |
| ["samples/construction_084.jpg", "realistic"], | |
| ["samples/construction_084.jpg", "futuristic"], | |
| ["samples/construction_084.jpg", "artistic"], | |
| ["samples/construction_085.jpg", "realistic"], | |
| ["samples/construction_085.jpg", "futuristic"], | |
| ["samples/construction_085.jpg", "artistic"], | |
| ["samples/construction_086.jpg", "realistic"], | |
| ["samples/construction_086.jpg", "futuristic"], | |
| ["samples/construction_086.jpg", "artistic"], | |
| ["samples/construction_087.jpg", "realistic"], | |
| ["samples/construction_087.jpg", "futuristic"], | |
| ["samples/construction_087.jpg", "artistic"], | |
| ["samples/road_088.jpg", "realistic"], | |
| ["samples/road_088.jpg", "futuristic"], | |
| ["samples/road_088.jpg", "artistic"], | |
| ["samples/road_089.jpg", "realistic"], | |
| ["samples/road_089.jpg", "futuristic"], | |
| ["samples/road_089.jpg", "artistic"], | |
| ["samples/road_090.jpg", "realistic"], | |
| ["samples/road_090.jpg", "futuristic"], | |
| ["samples/road_090.jpg", "artistic"], | |
| ["samples/construction_091.jpg", "realistic"], | |
| ["samples/construction_091.jpg", "futuristic"], | |
| ["samples/construction_091.jpg", "artistic"], | |
| ["samples/construction_092.jpg", "realistic"], | |
| ["samples/construction_092.jpg", "futuristic"], | |
| ["samples/construction_092.jpg", "artistic"], | |
| ["samples/construction_093.jpg", "realistic"], | |
| ["samples/construction_093.jpg", "futuristic"], | |
| ["samples/construction_093.jpg", "artistic"], | |
| ["samples/construction_094.jpg", "realistic"], | |
| ["samples/construction_094.jpg", "futuristic"], | |
| ["samples/construction_094.jpg", "artistic"], | |
| ["samples/construction_095.jpg", "realistic"], | |
| ["samples/construction_095.jpg", "futuristic"], | |
| ["samples/construction_095.jpg", "artistic"], | |
| ["samples/construction_096.jpg", "realistic"], | |
| ["samples/construction_096.jpg", "futuristic"], | |
| ["samples/construction_096.jpg", "artistic"], | |
| ["samples/construction_097.jpg", "realistic"], | |
| ["samples/construction_097.jpg", "futuristic"], | |
| ["samples/construction_097.jpg", "artistic"], | |
| ["samples/construction_098.jpg", "realistic"], | |
| ["samples/construction_098.jpg", "futuristic"], | |
| ["samples/construction_098.jpg", "artistic"], | |
| ["samples/construction_099.jpg", "realistic"], | |
| ["samples/construction_099.jpg", "futuristic"], | |
| ["samples/construction_099.jpg", "artistic"], | |
| ["samples/construction_100.jpg", "realistic"], | |
| ["samples/construction_100.jpg", "futuristic"], | |
| ["samples/construction_100.jpg", "artistic"], | |
| ["samples/construction_101.jpg", "realistic"], | |
| ["samples/construction_101.jpg", "futuristic"], | |
| ["samples/construction_101.jpg", "artistic"], | |
| ["samples/construction_102.jpg", "realistic"], | |
| ["samples/construction_102.jpg", "futuristic"], | |
| ["samples/construction_102.jpg", "artistic"], | |
| ["samples/construction_103.jpg", "realistic"], | |
| ["samples/construction_103.jpg", "futuristic"], | |
| ["samples/construction_103.jpg", "artistic"], | |
| ["samples/construction_104.jpg", "realistic"], | |
| ["samples/construction_104.jpg", "futuristic"], | |
| ["samples/construction_104.jpg", "artistic"], | |
| ["samples/construction_105.jpg", "realistic"], | |
| ["samples/construction_105.jpg", "futuristic"], | |
| ["samples/construction_105.jpg", "artistic"], | |
| ["samples/construction_106.jpg", "realistic"], | |
| ["samples/construction_106.jpg", "futuristic"], | |
| ["samples/construction_106.jpg", "artistic"], | |
| ["samples/construction_107.jpg", "realistic"], | |
| ["samples/construction_107.jpg", "futuristic"], | |
| ["samples/construction_107.jpg", "artistic"], | |
| ["samples/construction_108.jpg", "realistic"], | |
| ["samples/construction_108.jpg", "futuristic"], | |
| ["samples/construction_108.jpg", "artistic"], | |
| ["samples/construction_109.jpg", "realistic"], | |
| ["samples/construction_109.jpg", "futuristic"], | |
| ["samples/construction_109.jpg", "artistic"], | |
| ["samples/construction_110.jpg", "realistic"], | |
| ["samples/construction_110.jpg", "futuristic"], | |
| ["samples/construction_110.jpg", "artistic"], | |
| ["samples/construction_111.jpg", "realistic"], | |
| ["samples/construction_111.jpg", "futuristic"], | |
| ["samples/construction_111.jpg", "artistic"], | |
| ["samples/road_112.jpg", "realistic"], | |
| ["samples/road_112.jpg", "futuristic"], | |
| ["samples/road_112.jpg", "artistic"], | |
| ["samples/road_113.jpg", "realistic"], | |
| ["samples/road_113.jpg", "futuristic"], | |
| ["samples/road_113.jpg", "artistic"], | |
| ["samples/road_114.jpg", "realistic"], | |
| ["samples/road_114.jpg", "futuristic"], | |
| ["samples/road_114.jpg", "artistic"], | |
| ["samples/construction_115.jpg", "realistic"], | |
| ["samples/construction_115.jpg", "futuristic"], | |
| ["samples/construction_115.jpg", "artistic"], | |
| ["samples/construction_116.jpg", "realistic"], | |
| ["samples/construction_116.jpg", "futuristic"], | |
| ["samples/construction_116.jpg", "artistic"], | |
| ["samples/construction_117.jpg", "realistic"], | |
| ["samples/construction_117.jpg", "futuristic"], | |
| ["samples/construction_117.jpg", "artistic"], | |
| ], | |
| ["samples/1--9-_jpg.rf.9644d63e3fac251374ff5bcafcd46df6.jpg", "futuristic"], | |
| ["samples/1--9-_jpg.rf.550c50fd8d264a4635d969a3f6e58e20.jpg", "artistic"], | |
| ["samples/1--87-_jpg.rf.d182208e08a09865edf36470a91b59ee.jpg", "realistic"], | |
| ["samples/1--87-_jpg.rf.2f2c1509d7062a1ca7cf2e7f11524025.jpg", "futuristic"], | |
| ["samples/1--87-_jpg.rf.2573bc211e3ebd7dab8a8e2063f04e72.jpg", "artistic"], | |
| ["samples/1--86-_jpg.rf.d9851354896076145479a3255ad28983.jpg", "realistic"], | |
| ["samples/1--86-_jpg.rf.8743c06fb7171470aad8c29347787e02.jpg", "futuristic"], | |
| ["samples/1--86-_jpg.rf.31c349dab6d93647d16b644b5c01a701.jpg", "artistic"], | |
| ["samples/1--83-_jpg.rf.8eb1985273ed7e729ad48bbaa70b0a2c.jpg", "realistic"], | |
| ["samples/1--83-_jpg.rf.1e63ce7732332e644e18b5fc6a5d5b82.jpg", "futuristic"], | |
| ["samples/1--83-_jpg.rf.161813aa9068d2485cdd8fbf5aa235b6.jpg", "artistic"], | |
| ["samples/1--82-_jpg.rf.ed991a1e1ff5c52aae32d2b184e8879b.jpg", "realistic"], | |
| ["samples/1--82-_jpg.rf.afcf78cc953866c2b426d5d29177945b.jpg", "futuristic"], | |
| ["samples/1--82-_jpg.rf.a125202636d34fbf5f68bf9f169f9b45.jpg", "artistic"], | |
| # Road construction samples | |
| ["samples/1--80-_jpg.rf.d645bf21716e1036496ad924ed79ac6d.jpg", "realistic"], | |
| ["samples/1--80-_jpg.rf.87703d83ea49cb7af1cd234f0421d7da.jpg", "futuristic"], | |
| ["samples/1--80-_jpg.rf.75ef960e488a60fb649b2c6901691cc1.jpg", "artistic"], | |
| ["samples/1--76-_jpg.rf.cbf88ab7105bde653dee9be8c2264c5b.jpg", "realistic"], | |
| ["samples/1--76-_jpg.rf.6798acea648c368e35ef43590a374130.jpg", "futuristic"], | |
| ["samples/1--76-_jpg.rf.4e4e22f8d28e0db71791c0b878d4c2ec.jpg", "artistic"], | |
| # Bridge construction samples | |
| ["samples/1--73-_jpg.rf.b3be7aeb6c77ec6179b486044e900687.jpg", "realistic"], | |
| ["samples/1--73-_jpg.rf.56f7393e7b97bd978b02f550c2a8936b.jpg", "futuristic"], | |
| ["samples/1--73-_jpg.rf.067675d59f006f56190edd5939f6cbfb.jpg", "artistic"], | |
| ["samples/1--7-_jpg.rf.c4bca273346172de9cde286557f61e1d.jpg", "realistic"], | |
| ["samples/1--7-_jpg.rf.3f445c7a434d7d73680221c09300ea61.jpg", "futuristic"], | |
| ["samples/1--7-_jpg.rf.14efa5cd7f0a62e8fb7170a595110d57.jpg", "artistic"], | |
| # Additional construction samples | |
| ["samples/1--68-_jpg.rf.f793eec06843dbcd3712155c0c765e39.jpg", "realistic"], | |
| ["samples/1--68-_jpg.rf.84fafd8af20a11a1daca0749141cdff7.jpg", "futuristic"], | |
| ["samples/1--68-_jpg.rf.0df427eb151c1099ce8c1d56777b06e6.jpg", "artistic"], | |
| ["samples/1--67-_jpg.rf.ddd62d64ddd51eeb07c1de41304cd941.jpg", "realistic"], | |
| ["samples/1--67-_jpg.rf.bc2d043da983274e4d16632cdcd55a1e.jpg", "futuristic"], | |
| ["samples/1--67-_jpg.rf.6f758b1739647e24987418df0e3059fd.jpg", "artistic"], | |
| ["samples/1--65-_jpg.rf.a7039542180728c7cbb8bb5b423f9807.jpg", "realistic"], | |
| ["samples/1--65-_jpg.rf.82fc8f90e8d7536ae6c59f9c03141ef7.jpg", "futuristic"], | |
| ["samples/1--65-_jpg.rf.4a2d32c2552f489771be228141b226c7.jpg", "artistic"], | |
| ["samples/1--64-_jpg.rf.fc7a4fae31efc4a6a63879cb36a39c47.jpg", "realistic"], | |
| ["samples/1--64-_jpg.rf.4af108f9cac6e83adfc416659faf7848.jpg", "futuristic"], | |
| ["samples/1--64-_jpg.rf.2dacf22b827fcf1e2ca4e332c2abf9df.jpg", "artistic"], | |
| ["samples/1--63-_jpg.rf.4ecddff8f7fe664fc147931092790bd7.jpg", "realistic"], | |
| ["samples/1--63-_jpg.rf.37e85abf8d048ef9ce75d47210626718.jpg", "futuristic"], | |
| ["samples/1--63-_jpg.rf.32a6ec79dc25a0b176034fed6d2489d5.jpg", "artistic"], | |
| ["samples/1--62-_jpg.rf.c9c05a5fa96047ad20312ab59825f592.jpg", "realistic"], | |
| ["samples/1--62-_jpg.rf.681ae673b6af7ff87056809c706f4f35.jpg", "futuristic"], | |
| ["samples/1--62-_jpg.rf.34bffd3e20b20147dc9fabac7eb08c2a.jpg", "artistic"], | |
| ["samples/1--61-_jpg.rf.cf3a2daa89d567a47d5355bdba9a0198.jpg", "realistic"], | |
| ["samples/1--61-_jpg.rf.7a107c7913b54770d160d61a90f19f25.jpg", "futuristic"], | |
| ["samples/1--61-_jpg.rf.2bab5b0ba52975e1362a83fa06f71ad2.jpg", "artistic"], | |
| ["samples/1--60-_jpg.rf.e3d64315073b5d9bd72f6f9134dd00b6.jpg", "realistic"], | |
| ["samples/1--60-_jpg.rf.6415acdc3ffcd3dae52acb992308b271.jpg", "futuristic"], | |
| ["samples/1--60-_jpg.rf.34994f98c94dc9b6e092ed8d181e21b5.jpg", "artistic"], | |
| ["samples/1--54-_jpg.rf.ff0c9afdc5e574b293416150bd71da56.jpg", "realistic"], | |
| ["samples/1--54-_jpg.rf.d84d03f1614cd1478ef289b9970a7505.jpg", "futuristic"], | |
| ["samples/1--54-_jpg.rf.40f198aa313d81c2339ba76adec7b632.jpg", "artistic"], | |
| ["samples/1--52-_jpg.rf.f5b8f3320c7742a8563c94ef11ffe685.jpg", "realistic"], | |
| ["samples/1--52-_jpg.rf.f1fccdb171be2239a443a2770ed81cad.jpg", "futuristic"], | |
| ["samples/1--52-_jpg.rf.b433758070b56a1bb117913707164f73.jpg", "artistic"], | |
| ["samples/1--46-_jpg.rf.6d6fc570bf6f6c50fed1b0c1c1dc1f48.jpg", "realistic"], | |
| ["samples/1--46-_jpg.rf.497808af9bd186d92da07e3dd6ff339a.jpg", "futuristic"], | |
| ["samples/1--46-_jpg.rf.2edd9681323a6c6df913efe3ad937904.jpg", "artistic"], | |
| ["samples/1--41-_jpg.rf.ae0be14f17ed018288c6b2b0c21dc252.jpg", "realistic"], | |
| ["samples/1--41-_jpg.rf.2fa10765708519bb42363bbd2668d498.jpg", "futuristic"], | |
| ["samples/1--41-_jpg.rf.292b9eda9b4ab1ea6601d8af9a3fe1e6.jpg", "artistic"], | |
| ["samples/1--4-_jpg.rf.db90e4542575b4320aea6423998b87af.jpg", "realistic"], | |
| ["samples/1--4-_jpg.rf.643b5c3d9bde02cdebfb2c8f2d6760c2.jpg", "futuristic"], | |
| ["samples/1--4-_jpg.rf.0d1c2fe6d5869d80dbca50a038d6d2b8.jpg", "artistic"], | |
| ["samples/1--38-_jpg.rf.7ba6249e6836aef42805430984983714.jpg", "realistic"], | |
| ["samples/1--38-_jpg.rf.558bed32097bb8f0604b9ee9cc01c32c.jpg", "futuristic"], | |
| ["samples/1--38-_jpg.rf.40d2c933f09ee7356bc6f6f184f9077b.jpg", "artistic"], | |
| ["samples/1--37-_jpg.rf.5e1be607b6042a1b9fc508f9d6c711bf.jpg", "realistic"], | |
| ["samples/1--37-_jpg.rf.59de51b1dd967795475dc1c5b7926706.jpg", "futuristic"], | |
| ["samples/1--37-_jpg.rf.4ba07f4c07a3e70e420000f9b1c3dad6.jpg", "artistic"], | |
| ["samples/1--35-_jpg.rf.f3cc69fd984b810bd091efa6a4a16021.jpg", "realistic"], | |
| ["samples/1--35-_jpg.rf.3d52db41936513a4ca9b46341660ed0f.jpg", "futuristic"], | |
| ["samples/1--35-_jpg.rf.144b6d87f8b5a9af9a56e062de67b312.jpg", "artistic"], | |
| ["samples/1--33-_jpg.rf.d14e04d6803e79def00b1e030fa591eb.jpg", "realistic"], | |
| ["samples/1--33-_jpg.rf.8d1f8d5887f8796467a87415e214379f.jpg", "futuristic"], | |
| ["samples/1--33-_jpg.rf.156db5e009b3f61d12ff792f62f857b5.jpg", "artistic"], | |
| ["samples/1--32-_jpg.rf.a3da928423c53a107674eff5e93556db.jpg", "realistic"], | |
| ["samples/1--32-_jpg.rf.0ab6901ce392a13967333e0f229244c6.jpg", "futuristic"], | |
| ["samples/1--32-_jpg.rf.0a88170e18d027822d1e62164e197eff.jpg", "artistic"], | |
| ["samples/1--31-_jpg.rf.98ba9bf87ffdbd8ae22e56b3d7dbb2b7.jpg", "realistic"], | |
| ["samples/1--31-_jpg.rf.3a2df09b1b10348259a032a0a663267e.jpg", "futuristic"], | |
| ["samples/1--31-_jpg.rf.14a871bdca9c22f2aea1c697574175b3.jpg", "artistic"], | |
| ["samples/1--30-_jpg.rf.bd468ebdd124667612002bf0fb487905.jpg", "realistic"], | |
| ["samples/1--30-_jpg.rf.4fafec2b0bdbfb7a2c189fcbc0eb0fdd.jpg", "futuristic"], | |
| ["samples/1--30-_jpg.rf.27581192d14af812295a2bb70dd9b7a4.jpg", "artistic"], | |
| ["samples/1--3-_jpg.rf.c3ef6c743430bfd6e9aac8115389fed3.jpg", "realistic"], | |
| ["samples/1--3-_jpg.rf.c17eee911f46916ce26167de42ea5196.jpg", "futuristic"], | |
| ["samples/1--3-_jpg.rf.0e9c606a0bb6ada5bbc1b2fdb4ec8c9b.jpg", "artistic"], | |
| ["samples/1--27-_jpg.rf.a486cf9e45f8c7accc373780422ae975.jpg", "realistic"], | |
| ["samples/1--27-_jpg.rf.1ac5bbbb7bbf55d72b645979a6403a2e.jpg", "futuristic"], | |
| ["samples/1--27-_jpg.rf.0be78005ac20e89d83e76365676aaf3b.jpg", "artistic"], | |
| ["samples/1--24-_jpg.rf.c0b4a42c7f837b0870930f120efd17fe.jpg", "realistic"], | |
| ["samples/1--24-_jpg.rf.8b40905de0a5235d5ee3bf643c4caa2d.jpg", "futuristic"], | |
| ["samples/1--24-_jpg.rf.8861c49c2ab8560b25dcb5a428697485.jpg", "artistic"], | |
| ["samples/1--17-_jpg.rf.d13676a567fb592569d2290f0270f981.jpg", "realistic"], | |
| ["samples/1--17-_jpg.rf.b2cd1a3e657c0fde9a757c385234729c.jpg", "futuristic"], | |
| ["samples/1--17-_jpg.rf.7928e0354112b2c8901f0de35bf2ac65.jpg", "artistic"], | |
| ["samples/1--16-_jpg.rf.96a61df2434fc94c841c8e8c626eb208.jpg", "realistic"], | |
| # Include the original demo samples as well | |
| ["samples/building_construction.jpg", "realistic"], | |
| ["samples/bridge_construction.jpg", "futuristic"], | |
| ["samples/road_construction.jpg", "artistic"], | |
| ], | |
| inputs=[image_input, style_selector], | |
| label="Sample Construction Images - Click any example to try it!" | |
| ) | |
| # Event handlers | |
| process_btn.click( | |
| fn=process_image, | |
| inputs=[image_input, style_selector], | |
| outputs=[original_output, detected_output, completed_output, comparison, status_text, audio_output] | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch( | |
| server_name="0.0.0.0", | |
| server_port=7860, | |
| share=True, | |
| show_error=True | |
| ) | |