first commit

.gitignore

@@ -0,0 +1,198 @@
+# 环境配置文件
+.env
+.env.local
+.env.*.local
+
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+Pipfile.lock
+
+# PEP 582
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# IDEs
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+.DS_Store?
+._*
+.Spotlight-V100
+.Trashes
+ehthumbs.db
+Thumbs.db
+
+# 数据库文件
+*.db
+*.sqlite
+*.sqlite3
+
+# 日志文件
+*.log
+logs/
+
+# 临时文件
+tmp/
+temp/
+*.tmp
+
+# SSL证书文件
+*.pem
+*.crt
+*.key
+
+# 敏感信息目录
+secrets/
+
+# 上传文件
+uploads/
+static/uploads/
+
+# 临时调试文件和脚本
+debug_*.py
+test_*_fix.py
+test_*_cache*.py
+simple_cache_test.py
+*_debug_report_*.md
+fix_*.py
+*_investigation.py
+*_diagnosis.py
+*_debug_temp.py
+trace_*.py
+temp_*.py
+
+# 临时脚本和报告
+security_audit_report.txt
+gradio_debug_report_*.md
+backend_test_results.json
+final_validation_summary.py
+*.investigation
+*.diagnosis
+
+# 但保留项目中的重要测试文件
+!tests/
+!fastapi-backend/tests/
+!**/test_*.py
+!simple_gradio_app.py
+!verify_*.py
+!check_*.py
+
+# Gradio cache and generated files
+.gradio/
+gradio_cached_examples/

config.py

@@ -0,0 +1,64 @@
+import os
+from typing import Final
+from pydantic import BaseModel, Field
+
+from dotenv import load_dotenv
+load_dotenv()
+
+class AppConfig(BaseModel):
+    API_BASE_URL: str = Field(default=os.getenv("API_BASE_URL", "http://localhost:8000"), description="Backend API base URL")
+    API_AUTH_TOKEN: str = Field(default=os.getenv("API_AUTH_TOKEN", ""), description="API authentication token - REQUIRED")
+    API_TIMEOUT: int = Field(default=int(os.getenv("API_TIMEOUT", "120")), description="API request timeout in seconds")
+
+    def model_post_init(self, __context) -> None:
+    
+        if not self.API_AUTH_TOKEN:
+            raise ValueError("❌ API_AUTH_TOKEN 环境变量必须设置")
+
+        # 简化的禁用令牌检查
+        forbidden_tokens = {
+            "",
+            "your_secret_token_here",
+            "your_api_token_here"
+        }
+
+        if self.API_AUTH_TOKEN in forbidden_tokens:
+            raise ValueError(f"❌ 禁止使用占位符令牌: {self.API_AUTH_TOKEN}")
+
+        # 简化：只要求基本长度
+        if len(self.API_AUTH_TOKEN) < 8:
+            raise ValueError("❌ 认证令牌长度必须至少8字符")
+    
+    # Polling Configuration
+    POLL_INTERVAL: float = Field(default=3.0, description="Status polling interval in seconds")
+    MAX_POLL_ATTEMPTS: int = Field(default=900, description="Maximum polling attempts (45 minutes for new workflow)")
+    
+    # UI Configuration
+    APP_TITLE: str = Field(default="AI Image Generator", description="Application title")
+    APP_DESCRIPTION: str = Field(default="Robust AI image generation with clean state management", description="App description")
+
+    # Server Configuration
+    SERVER_PORT: int = Field(default=int(os.getenv("SERVER_PORT", "7860")), description="Server port")
+    ENABLE_SHARE: bool = Field(default=os.getenv("ENABLE_SHARE", "false").lower() == "true", description="Enable Gradio sharing")
+
+    # Authentication Configuration
+    ENABLE_AUTH: bool = Field(default=os.getenv("ENABLE_AUTH", "false").lower() == "true", description="Enable authentication")
+    AUTH_USERNAME: str = Field(default=os.getenv("AUTH_USERNAME", "admin"), description="Authentication username")
+    AUTH_PASSWORD: str = Field(default=os.getenv("AUTH_PASSWORD", "password"), description="Authentication password")
+
+    # Logging Configuration
+    LOG_LEVEL: str = Field(default="WARNING", description="Logging level")
+    ENABLE_DEBUG_LOGGING: bool = Field(default=False, description="Enable detailed debug logging")
+
+CONFIG: Final[AppConfig] = AppConfig()
+
+def get_api_headers() -> dict[str, str]:
+
+    return {
+        "Content-Type": "application/json",
+        "X-API-Auth": CONFIG.API_AUTH_TOKEN
+    }
+
+def get_api_url(endpoint: str) -> str:
+
+    return f"{CONFIG.API_BASE_URL.rstrip('/')}/{endpoint.lstrip('/')}"

examples_config.py

@@ -0,0 +1,502 @@
+TEXT_TO_IMAGE_EXAMPLES = [
+    ["An orange cat sitting on a windowsill, sunlight streaming through the window onto it, cozy home environment, high-definition photography"],
+    ["A majestic tiger walking through the jungle, sharp eyes, clear fur details, wildlife photography"],
+    ["Futuristic tech city at night, neon lights flashing, flying cars shuttling through the air, cyberpunk style, 4K quality"],
+    ["Steampunk-style robot, intricate gears and steam, retro industrial design, metallic texture"],
+    ["Beautiful woman in traditional Chinese clothing playing guqin in bamboo forest, elegant atmosphere, Chinese painting style, fine brushwork"],
+    ["Japanese garden with cherry blossoms falling, ancient wooden buildings, stone lanterns, zen aesthetics, spring"],
+    ["A castle floating in the clouds, surrounded by rainbows and flying birds, dreamy fairy tale style, Disney animation"],
+    ["Fairy in magical forest, glowing wings, mysterious mushroom house, fantasy illustration style"],
+
+    # Natural Landscape
+    ["Peaceful lake reflecting snow-capped mountains, sunrise moment, golden sunlight on water surface, landscape photography"],
+    ["Mysterious creatures in the deep sea, glowing jellyfish and coral, blue dreamy lighting, ocean wonder"],
+
+    # Artistic Creative
+    ["Abstract art style city buildings, geometric combinations, vibrant color contrasts, modern art"],
+    ["Vaporwave aesthetic sunset, pink-purple tones, retro grid background, 80s nostalgia"]
+]
+
+TEXT_TO_IMAGE_EXAMPLES_WITH_RESULTS = [
+    # 格式: [提示词, 分辨率, 图片文件路径]
+    # Animals
+    [
+        "An orange cat sitting on a windowsill, sunlight streaming through the window onto it, cozy home environment, high-definition photography",
+        "square - 1024x1024 (1:1)",
+        "examples/results/cat_windowsill.jpg"
+    ],
+    [
+        "A majestic tiger walking through the jungle, sharp eyes, clear fur details, wildlife photography",
+        "landscape - 1152x896 (4:3)",
+        "examples/results/tiger_jungle.jpg"
+    ],
+
+    # Sci-fi Future
+    [
+        "Futuristic tech city at night, neon lights flashing, flying cars shuttling through the air, cyberpunk style, 4K quality",
+        "landscape - 1344x768 (16:9)",
+        "examples/results/cyberpunk_city.jpg"
+    ],
+    [
+        "Steampunk-style robot, intricate gears and steam, retro industrial design, metallic texture",
+        "square - 1024x1024 (1:1)",
+        "examples/results/steampunk_robot.jpg"
+    ],
+
+    # Traditional Oriental
+    [
+        "Beautiful woman in traditional Chinese clothing playing guqin in bamboo forest, elegant atmosphere, Chinese painting style, fine brushwork",
+        "portrait - 768x1344 (9:16)",
+        "examples/results/chinese_woman_guqin.jpg"
+    ],
+    [
+        "Japanese garden with cherry blossoms falling, ancient wooden buildings, stone lanterns, zen aesthetics, spring",
+        "landscape - 1216x832 (3:2)",
+        "examples/results/japanese_garden.jpg"
+    ],
+
+    # Fantasy Fairy Tale
+    [
+        "A castle floating in the clouds, surrounded by rainbows and flying birds, dreamy fairy tale style, Disney animation",
+        "landscape - 1152x896 (4:3)",
+        "examples/results/floating_castle.jpg"
+    ],
+    [
+        "Fairy in magical forest, glowing wings, mysterious mushroom house, fantasy illustration style",
+        "square - 1024x1024 (1:1)",
+        "examples/results/fairy_forest.jpg"
+    ],
+
+    # Natural Landscape
+    [
+        "Peaceful lake reflecting snow-capped mountains, sunrise moment, golden sunlight on water surface, landscape photography",
+        "landscape - 1344x768 (16:9)",
+        "examples/results/mountain_lake.jpg"
+    ],
+    [
+        "Mysterious creatures in the deep sea, glowing jellyfish and coral, blue dreamy lighting, ocean wonder",
+        "portrait - 896x1152 (3:4)",
+        "examples/results/deep_sea_creatures.jpg"
+    ],
+
+    # Artistic Creative
+    [
+        "Abstract art style city buildings, geometric combinations, vibrant color contrasts, modern art",
+        "square - 1024x1024 (1:1)",
+        "examples/results/abstract_buildings.jpg"
+    ],
+    [
+        "Vaporwave aesthetic sunset, pink-purple tones, retro grid background, 80s nostalgia",
+        "landscape - 1536x640 (21:9)",
+        "examples/results/vaporwave_sunset.jpg"
+    ]
+]
+
+# 文生图示例标签（可选，用于更好的用户体验）
+TEXT_TO_IMAGE_LABELS = [
+    "🐱 Cozy Cat",
+    "🐅 Wild Tiger",
+    "🌃 Cyberpunk City",
+    "🤖 Steampunk Robot",
+    "🎵 Elegant Guqin Lady",
+    "🌸 Japanese Sakura Garden",
+    "🏰 Castle in the Clouds",
+    "🧚 Forest Fairy",
+    "🏔️ Sunrise over Snowy Lake",
+    "🌊 Deep Sea Wonders",
+    "🏙️ Abstract Architecture",
+    "🌅 Vaporwave Sunset"
+]
+
+# 按类别组织的示例（用于高级展示）
+CATEGORIZED_EXAMPLES = {
+    "🐾 Animal World": [
+        ["An orange cat sitting on a windowsill, sunlight streaming through the window onto it, cozy home environment, high-definition photography"],
+        ["A majestic tiger walking through the jungle, sharp eyes, clear fur details, wildlife photography"],
+        ["An elegant white swan swimming on the lake, feathers white as snow, clear reflection, nature photography"]
+    ],
+
+    "🚀 Sci-Fi Future": [
+        ["Futuristic tech city at night, neon lights flashing, flying cars shuttling through the air, cyberpunk style, 4K quality"],
+        ["Steampunk-style robot, intricate gears and steam, retro industrial design, metallic texture"],
+        ["Inside a space station, astronauts floating in zero gravity, sci-fi movie scene, high-tech equipment"]
+    ],
+
+    "🏮 Traditional Aesthetics": [
+        ["Beautiful woman in traditional Chinese clothing playing guqin in bamboo forest, elegant atmosphere, Chinese painting style, fine brushwork"],
+        ["Japanese garden with cherry blossoms falling, ancient wooden buildings, stone lanterns, zen aesthetics, spring"],
+        ["Traditional Chinese architecture complex, flying eaves, red walls and yellow tiles, architectural photography"]
+    ],
+
+    "✨ Fantasy Magic": [
+        ["A castle floating in the clouds, surrounded by rainbows and flying birds, dreamy fairy tale style, Disney animation"],
+        ["Fairy in magical forest, glowing wings, mysterious mushroom house, fantasy illustration style"],
+        ["Dragon hovering over mountain peaks, breathing fire, epic fantasy scene, cinematic quality"]
+    ],
+
+    "🌄 Natural Landscape": [
+        ["Peaceful lake reflecting snow-capped mountains, sunrise moment, golden sunlight on water surface, landscape photography"],
+        ["Mysterious creatures in the deep sea, glowing jellyfish and coral, blue dreamy lighting, ocean wonder"],
+        ["Autumn maple forest with red and yellow leaves, sunlight filtering through leaves, warm tones"]
+    ]
+}
+
+# 图生图基本信息
+IMAGE_TO_IMAGE_INFO = {
+    "supported_formats": ["PNG", "JPEG", "BMP", "WEBP"],
+    "file_size_limit": "10MB"
+}
+
+# 摄影风格转换基本信息
+PHOTO_STYLE_INFO = {
+    "supported_formats": ["PNG", "JPEG", "BMP", "WEBP"],
+    "file_size_limit": "10MB",
+    "max_resolution": "2000x2000",
+    "min_resolution": "500x500",
+    "auto_resize": True,
+    "description": "Transform your photos with professional photography styles"
+}
+
+# 摄影风格列表和描述
+PHOTO_STYLES = [
+    "Photo-Camera Movement",                    # 移动镜头
+    "Photo-Relighting",                        # 重新布光
+    "Photo-Camera Zoom",                       # 画面缩放
+    "Photo-Professional Product Photography",   # 专业产品图
+    "Photo-Tilt-Shift Miniature",             # 微缩世界
+    "Photo-Reflection Addition",               # 添加倒影
+    "Photo-Character Pose & Viewpoint Change"  # 角色姿势视角变换
+]
+
+PHOTO_STYLE_DESCRIPTIONS = {
+    "Photo-Camera Movement": {
+        "name": "Camera Movement",
+        "description": "Create dynamic camera motion effects to enhance movement and visual impact",
+        "effect": "Best for sports scenes, street photography, dynamic portraits",
+        "example": "Turn static photos into motion trail and motion blur effects"
+    },
+    "Photo-Relighting": {
+        "name": "Relighting",
+        "description": "Modify the lighting and atmosphere of the photo, optimizing light and shadow relationships",
+        "effect": "Great for portrait, indoor scenes, and product photography",
+        "example": "Adjust light direction, intensity, and color temperature to create different visual moods"
+    },
+    "Photo-Camera Zoom": {
+        "name": "Camera Zoom",
+        "description": "Adjust zoom and focus effects to highlight the subject or change perspective",
+        "effect": "Suitable for landscapes, architecture, and close-ups",
+        "example": "Create telephoto compression or wide-angle perspective effects"
+    },
+    "Photo-Professional Product Photography": {
+        "name": "Professional Product Photography",
+        "description": "Convert to professional product photography style, highlighting product features",
+        "effect": "Ideal for e-commerce, industrial design, and showcases",
+        "example": "Optimize lighting, background, and composition to increase commercial value"
+    },
+    "Photo-Tilt-Shift Miniature": {
+        "name": "Tilt-Shift Miniature",
+        "description": "Create a miniature model-like effect for a whimsical atmosphere",
+        "effect": "Works well for cityscapes, building clusters, and top-down shots",
+        "example": "Use selective focus and color tuning to make real scenes look like models"
+    },
+    "Photo-Reflection Addition": {
+        "name": "Reflection Addition",
+        "description": "Add natural reflections to images to enhance visual layering",
+        "effect": "Good for water surfaces, glass, and mirror effects",
+        "example": "Add realistic reflections on water, ground, or other reflective surfaces"
+    },
+    "Photo-Character Pose & Viewpoint Change": {
+        "name": "Pose & Viewpoint Change",
+        "description": "Change human pose and shooting angle to create new compositions",
+        "effect": "Useful for portrait, fashion, and artistic photography",
+        "example": "Adjust body movements, expressions, and camera angle for better visual results"
+    }
+}
+
+# 摄影风格转换示例配置
+PHOTO_STYLE_EXAMPLES = [
+    {
+        "style": "Photo-Camera Movement",
+        "description": "Add dynamic camera movement to static portraits",
+        "tags": ["dynamic", "motion", "street"]
+    },
+    {
+        "style": "Photo-Relighting",
+        "description": "Optimize lighting for indoor portraits",
+        "tags": ["portrait", "lighting", "ambience"]
+    },
+    {
+        "style": "Photo-Camera Zoom",
+        "description": "Adjust focal length effects for landscape photos",
+        "tags": ["landscape", "focal", "perspective"]
+    },
+    {
+        "style": "Photo-Professional Product Photography",
+        "description": "Convert regular product photos into professional commercial shots",
+        "tags": ["product", "commercial", "e-commerce"]
+    },
+    {
+        "style": "Photo-Tilt-Shift Miniature",
+        "description": "Turn cityscapes into miniature model effects",
+        "tags": ["miniature", "whimsical", "creative"]
+    },
+    {
+        "style": "Photo-Reflection Addition",
+        "description": "Add water reflections to architecture photos",
+        "tags": ["reflection", "water", "symmetry"]
+    },
+    {
+        "style": "Photo-Character Pose & Viewpoint Change",
+        "description": "Adjust pose and viewpoint for portrait photography",
+        "tags": ["portrait", "pose", "viewpoint"]
+    }
+]
+
+# 获取随机示例（用于动态展示）
+def get_random_examples(category=None, count=4):
+
+    import random
+    
+    if category and category in CATEGORIZED_EXAMPLES:
+        examples = CATEGORIZED_EXAMPLES[category]
+    else:
+        examples = TEXT_TO_IMAGE_EXAMPLES
+    
+    if len(examples) <= count:
+        return examples
+    
+    return random.sample(examples, count)
+
+# 获取分类示例（用于标签页展示）
+def get_categorized_examples():
+
+    return CATEGORIZED_EXAMPLES
+
+# 室内设计渲染示例配置
+
+# 室内设计渲染示例配置（重新实现）
+
+INTERIOR_DESIGN_EXAMPLES_WITH_RESULTS = [
+    [
+        "examples/interior_input.png",                          # 输入的白膜图
+        "japanese_wabi_sabi",                                   # 日本诧寂风格
+        "examples/results/interior_japanese_wabi_sabi.jpg"      # 渲染结果
+    ],
+    [
+        "examples/interior_input.png",                          # 输入的白膜图
+        "nordic_cozy",                                          # 北欧温馨风格
+        "examples/results/interior_nordic_cozy.jpg"             # 渲染结果
+    ],
+    [
+        "examples/interior_input.png",                          # 输入的白膜图
+        "italian_luxury",                                       # 意大利风格
+        "examples/results/interior_italian_luxury.jpg"          # 渲染结果
+    ],
+    [
+        "examples/interior_input.png",                          # 输入的白膜图
+        "parisian_apartment",                                   # 巴黎套间风格
+        "examples/results/interior_parisian_apartment.jpg"      # 渲染结果
+    ]
+]
+
+# 摄影风格转换相关辅助函数
+def get_photo_styles():
+
+    return PHOTO_STYLES.copy()
+
+def get_photo_style_description(style):
+
+    return PHOTO_STYLE_DESCRIPTIONS.get(style)
+
+def get_photo_style_info():
+
+    return PHOTO_STYLE_INFO.copy()
+
+def get_photo_style_examples():
+
+    return PHOTO_STYLE_EXAMPLES.copy()
+
+def validate_photo_style(style):
+
+    return style in PHOTO_STYLES
+
+def get_photo_styles_with_descriptions():
+
+    return PHOTO_STYLE_DESCRIPTIONS.copy()
+
+def get_random_photo_style_examples(count=3):
+
+    import random
+
+    if len(PHOTO_STYLE_EXAMPLES) <= count:
+        return PHOTO_STYLE_EXAMPLES.copy()
+
+    return random.sample(PHOTO_STYLE_EXAMPLES, count)
+
+# 线稿转换示例配置
+
+# 线稿转换基本信息
+LINE_ART_CONVERSION_INFO = {
+    "supported_formats": ["PNG", "JPEG", "BMP", "WEBP"],
+    "file_size_limit": "10MB",
+    "max_resolution": "2000x2000",
+    "min_resolution": "500x500",
+    "auto_resize": True,
+    "description": "Convert your photos into clean line art drawings"
+}
+
+# 线稿转换示例配置（简化格式，只包含输入和输出）
+# 格式: [输入图片路径, 结果图片路径]
+LINE_ART_CONVERSION_EXAMPLES_WITH_RESULTS = [
+    [
+        "examples/line_art_input/example1.jpg",                    # 输入图片
+        "examples/results/line_art_example1.jpg"                   # 结果图片
+    ],
+    [
+        "examples/line_art_input/example2.jpg",                    # 输入图片
+        "examples/results/line_art_example2.jpg"                   # 结果图片
+    ],
+    [
+        "examples/line_art_input/example3.jpg",                    # 输入图片
+        "examples/results/line_art_example3.jpg"                   # 结果图片
+    ]
+]
+
+# 线稿转换相关辅助函数
+def get_line_art_conversion_info():
+
+    return LINE_ART_CONVERSION_INFO.copy()
+
+def get_line_art_conversion_examples():
+
+    return LINE_ART_CONVERSION_EXAMPLES_WITH_RESULTS.copy()
+
+# 二次元转真人相关配置
+ANIME_TO_REAL_INFO = {
+    "description": "Convert anime illustrations into realistic photos",
+    "supported_formats": ["PNG", "JPEG", "JPG", "WEBP"],
+    "max_file_size": "10MB",
+    "recommended_resolution": "512x512 to 1024x1024",
+    "processing_time": "Typically 1–3 minutes"
+}
+
+# 二次元转真人示例配置
+
+ANIME_TO_REAL_EXAMPLES_WITH_RESULTS = [
+    ["examples/anime_input/example1.jpg","examples/results/anime_to_real_example1.jpg"]
+]
+
+# 用于显示预生成结果的映射（如果需要）
+ANIME_TO_REAL_RESULTS_MAP = {
+    "examples/anime_input/example1.jpg": "examples/results/anime_to_real_example1.jpg"
+}
+
+# 二次元转真人相关辅助函数
+def get_anime_to_real_info():
+
+    return ANIME_TO_REAL_INFO.copy()
+
+def get_anime_to_real_examples():
+
+    return ANIME_TO_REAL_EXAMPLES_WITH_RESULTS.copy()
+
+def get_random_line_art_examples(count=3):
+
+    import random
+
+    if len(LINE_ART_CONVERSION_EXAMPLES_WITH_RESULTS) <= count:
+        return LINE_ART_CONVERSION_EXAMPLES_WITH_RESULTS.copy()
+
+    return random.sample(LINE_ART_CONVERSION_EXAMPLES_WITH_RESULTS, count)
+
+# 真实图片转动漫插画相关配置
+REAL_TO_ANIME_INFO = {
+    "description": "Convert real photos into anime illustration style",
+    "supported_formats": ["PNG", "JPEG", "JPG", "WEBP"],
+    "max_file_size": "10MB",
+    "recommended_resolution": "512x512 to 1024x1024",
+    "processing_time": "Typically 1–3 minutes"
+}
+
+# 真实图片转动漫示例配置
+# 格式: [输入图片路径, 结果图片路径]
+REAL_TO_ANIME_EXAMPLES_WITH_RESULTS = [
+    ["examples/real_input/example1.jpg", "examples/results/real_to_anime_example1.jpg"],
+    ["examples/real_input/example2.jpg", "examples/results/real_to_anime_example2.jpg"],
+    ["examples/real_input/example3.jpg", "examples/results/real_to_anime_example3.jpg"]
+]
+
+# 用于显示预生成结果的映射（如果需要）
+REAL_TO_ANIME_RESULTS_MAP = {
+    "examples/real_input/example1.jpg": "examples/results/real_to_anime_example1.jpg",
+    "examples/real_input/example2.jpg": "examples/results/real_to_anime_example2.jpg",
+    "examples/real_input/example3.jpg": "examples/results/real_to_anime_example3.jpg"
+}
+
+# 真实图片转动漫相关辅助函数
+def get_real_to_anime_info():
+
+    return REAL_TO_ANIME_INFO.copy()
+
+def get_real_to_anime_examples():
+
+    return REAL_TO_ANIME_EXAMPLES_WITH_RESULTS.copy()
+
+def get_random_real_to_anime_examples(count=3):
+
+    import random
+
+    if len(REAL_TO_ANIME_EXAMPLES_WITH_RESULTS) <= count:
+        return REAL_TO_ANIME_EXAMPLES_WITH_RESULTS.copy()
+
+    return random.sample(REAL_TO_ANIME_EXAMPLES_WITH_RESULTS, count)
+
+FIVE_VIEW_GENERATION_EXAMPLES_WITH_RESULTS = [
+    # 格式: [输入图片路径, 结果图片路径]
+    [
+        "examples/five_view_input/example1.jpg",
+        "examples/results/five_view_example1.jpg"
+    ]
+]
+
+def get_random_five_view_examples(count=1):
+
+    import random
+
+    if len(FIVE_VIEW_GENERATION_EXAMPLES_WITH_RESULTS) <= count:
+        return FIVE_VIEW_GENERATION_EXAMPLES_WITH_RESULTS.copy()
+
+    return random.sample(FIVE_VIEW_GENERATION_EXAMPLES_WITH_RESULTS, count)
+
+# 图像扩展示例配置 - 使用多样化图片和真实AI生成的扩图结果
+IMAGE_OUTPAINTING_EXAMPLES_WITH_RESULTS = [
+
+    [
+        "examples/outpainting_input/example1.jpg",
+        0.2,  # 轻度扩展 - 原始测试图片
+        0.3,  # 中等宽度扩展
+        "examples/results/outpainting_example1.jpg"  # 真实AI生成结果
+    ],
+    [
+        "examples/outpainting_input/example2.jpg",
+        0.2,  # 轻度高度扩展 - 日式住宅，适合横向扩展
+        0.4,  # 较大宽度扩展
+        "examples/results/outpainting_example2.jpg"  # 待生成真实AI结果
+    ],
+    [
+        "examples/outpainting_input/example3.jpg",
+        0.5,  # 较大高度扩展 - 不露脸自拍，适合纵向扩展
+        0.2,  # 轻度宽度扩展
+        "examples/results/outpainting_example3.jpg"  # 待生成真实AI结果
+    ]
+]
+
+def get_random_outpainting_examples(count=3):
+
+    import random
+
+    if len(IMAGE_OUTPAINTING_EXAMPLES_WITH_RESULTS) <= count:
+        return IMAGE_OUTPAINTING_EXAMPLES_WITH_RESULTS.copy()
+
+    return random.sample(IMAGE_OUTPAINTING_EXAMPLES_WITH_RESULTS, count)
+

models.py

@@ -0,0 +1,200 @@
+from enum import Enum
+from typing import Optional, Any
+from datetime import datetime
+from pydantic import BaseModel, Field, validator
+import uuid
+
+class TaskStatus(str, Enum):
+
+    QUEUED = "queued"
+    PROCESSING = "processing"
+    COMPLETED = "completed"
+    FAILED = "failed"
+    CANCELLED = "cancelled"
+
+class UIState(str, Enum):
+
+    IDLE = "idle"
+    SUBMITTING = "submitting"
+    QUEUED = "queued"
+    PROCESSING = "processing"
+    COMPLETED = "completed"
+    ERROR = "error"
+
+class TaskInfo(BaseModel):
+
+    task_id: str = Field(..., description="Unique task identifier")
+    status: TaskStatus = Field(..., description="Current task status")
+    position: int = Field(default=0, description="Queue position")
+    message: str = Field(default="", description="Status message")
+    result_url: Optional[str] = Field(default=None, description="Result image URL")
+    error_message: Optional[str] = Field(default=None, description="Error message if failed")
+    
+    @validator('task_id')
+    def validate_task_id(cls, v):
+    
+        try:
+            uuid.UUID(v)
+            return v
+        except ValueError:
+            raise ValueError(f"Invalid task ID format: {v}")
+    
+    class Config:
+        frozen = True  # Immutable
+
+class UIStateInfo(BaseModel):
+
+    state: UIState = Field(..., description="Current UI state")
+    task_id: Optional[str] = Field(default=None, description="Associated task ID")
+    status_text: str = Field(default="", description="Display status text")
+    progress_text: str = Field(default="", description="Progress description")
+    show_image: bool = Field(default=False, description="Whether to show result image")
+    image_url: Optional[str] = Field(default=None, description="Result image URL")
+    submit_enabled: bool = Field(default=True, description="Submit button enabled")
+    cancel_enabled: bool = Field(default=False, description="Cancel button enabled")
+    timestamp: datetime = Field(default_factory=datetime.now, description="State timestamp")
+    
+    class Config:
+        frozen = True  # Immutable
+
+class TaskSubmission(BaseModel):
+
+    prompt: str = Field(..., min_length=5, max_length=4096, description="Image generation prompt")
+    width: int = Field(default=1024, description="Image width")
+    height: int = Field(default=1024, description="Image height")
+    task_type: str = Field(default="text_to_image", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "task_type": self.task_type,
+            "input_params": {
+                "prompt": self.prompt,
+                "width": self.width,
+                "height": self.height
+            }
+        }
+
+class ImageToImageSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64 encoded image data")
+    task_type: str = Field(default="image_to_image", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "image_data": self.image_data
+        }
+
+class PhotoStyleSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64 encoded image data")
+    style_preset: str = Field(..., description="Photography style preset")
+    task_type: str = Field(default="photo_style_transfer", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "image_data": self.image_data,
+            "style_preset": self.style_preset
+        }
+
+class InteriorDesignRenderingSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64 encoded white model image data")
+    design_style: str = Field(..., description="Interior design style")
+    task_type: str = Field(default="interior_design_rendering", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "image_data": self.image_data,
+            "design_style": self.design_style
+        }
+
+class WatermarkRemovalSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64 encoded image data")
+    task_type: str = Field(default="watermark_removal", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "image_data": self.image_data
+        }
+
+class LineArtConversionSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64 encoded image data")
+    task_type: str = Field(default="line_art_conversion", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "image_data": self.image_data
+        }
+
+class AnimeToRealSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64 encoded anime image data")
+    task_type: str = Field(default="anime_to_real", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "image_data": self.image_data
+        }
+
+class RealToAnimeSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64 encoded real image data")
+    task_type: str = Field(default="real_to_anime", description="Task type")
+
+    def to_api_payload(self) -> dict[str, Any]:
+    
+        return {
+            "image_data": self.image_data
+        }
+
+class AppSession(BaseModel):
+
+    session_id: str = Field(default_factory=lambda: str(uuid.uuid4()), description="Session ID")
+    current_ui_state: UIStateInfo = Field(default_factory=lambda: UIStateInfo(state=UIState.IDLE), description="Current UI state")
+    active_task_id: Optional[str] = Field(default=None, description="Currently active task ID")
+    last_submission: Optional[TaskSubmission] = Field(default=None, description="Last task submission")
+    created_at: datetime = Field(default_factory=datetime.now, description="Session creation time")
+    
+    def is_active_task(self, task_id: str) -> bool:
+    
+        return self.active_task_id == task_id
+    
+    def can_submit_new_task(self) -> bool:
+    
+        return self.current_ui_state.state in {UIState.IDLE, UIState.COMPLETED, UIState.ERROR}
+    
+    class Config:
+        frozen = False  # Mutable for state updates
+
+class ImageOutpaintingSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64编码的图片数据")
+    expand_height: float = Field(default=0.2, description="扩展高度百分比")
+    expand_width: float = Field(default=0.2, description="扩展宽度百分比")
+
+    def to_api_payload(self) -> dict:
+    
+        return {
+            "image_data": self.image_data,
+            "expand_height": self.expand_height,
+            "expand_width": self.expand_width
+        }
+
+class FiveViewGenerationSubmission(BaseModel):
+
+    image_data: str = Field(..., description="Base64编码的图片数据")
+
+    def to_api_payload(self) -> dict:
+    
+        return {
+            "image_data": self.image_data
+        }

requirements.txt

@@ -0,0 +1,6 @@
+# Gradio Frontend Requirements
+gradio[oauth]>=4.0.0
+httpx>=0.25.0
+Pillow>=9.0.0
+pydantic>=2.0.0
+python-dotenv>=1.0.0