Update app.py

app.py

@@ -18,83 +18,56 @@ pipe = DiffusionPipeline.from_pretrained(
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-# 플로우차트 예시
+# 인포그래픽 예시들
 EXAMPLES = [
     {
-        "title": "Business Workflow",
-        "prompt": """A hand-drawn style flowchart, vibrant colors, minimalistic icons.
-        BUSINESS WORKFLOW
-        ├── START [Green Button ~40px]
-        │   ├── COLLECT REQUIREMENTS [Folder Icon]
-        │   └── ANALYZE DATA [Chart Icon]
-        ├── IMPLEMENTATION [Coding Symbol ~50px]
-        │   ├── FRONTEND [Browser Icon]
-        │   └── BACKEND [Server Icon]
-        ├── TEST & INTEGRATION [Gear Icon ~45px]
-        └── DEPLOY
-            └── END [Checkered Flag ~40px]""",
+        "title": "Global Energy Mix",
+        "prompt": """A modern flat-style infographic about global renewable energy usage:
+        - Title: "Global Renewable Energy Share"
+        - Bar chart showing Solar, Wind, Hydro
+        - Pie chart: 40% Solar, 35% Wind, 25% Hydro
+        - Minimalist icons: sun, wind turbine, water wave
+        - Clean layout with pastel color scheme""",
         "width": 1024,
         "height": 1024
     },
     {
-        "title": "Software Release Flow",
-        "prompt": """A hand-drawn style flowchart, pastel colors, arrows between stages.
-        SOFTWARE RELEASE
-        ├── FEATURE BRANCH [Git Branch Icon ~45px]
-        │   ├── DEVELOPMENT [Code Editor]
-        │   └── UNIT TEST [Check Mark]
-        ├── MERGE TO MAIN [Pull Request Icon]
-        │   ├── CI/CD [Pipeline Icon ~40px]
-        │   └── BUILD [Gear Icon]
-        └── PRODUCTION
-            └── DEPLOY [Cloud Upload Icon]""",
+        "title": "Startup Growth Stats",
+        "prompt": """A hand-drawn style infographic illustrating startup growth stages:
+        - Title: "Startup Growth Journey"
+        - 5 main steps: Ideation, MVP, Early Users, Funding, Scaling
+        - Icons for each step (lightbulb, prototype gear, user group, money bag, rocket)
+        - Vibrant color palette""",
         "width": 1024,
         "height": 1024
     },
     {
-        "title": "E-Commerce Checkout",
-        "prompt": """A hand-drawn style flowchart, light watercolor, user journey from cart to payment.
-        E-COMMERCE CHECKOUT
-        ├── CART [Shopping Cart ~40px]
-        │   ├── LOGIN [User Icon]
-        │   └── ADDRESS [Location Pin]
-        ├── PAYMENT [Credit Card Icon ~45px]
-        │   ├── VALIDATION [Lock Icon]
-        │   └── CONFIRMATION [Receipt Icon]
-        └── ORDER COMPLETE
-            └── THANK YOU [Smiley Icon]""",
+        "title": "Healthy Lifestyle",
+        "prompt": """A watercolor style infographic showing healthy lifestyle tips:
+        - Title: "Healthy Lifestyle"
+        - 3 pillars: Nutrition, Exercise, Mental Health
+        - Icons: Food plate, running shoe, yoga person
+        - Soft pastel colors, gentle shapes""",
         "width": 1024,
         "height": 1024
     },
     {
-        "title": "Data Pipeline",
-        "prompt": """A hand-drawn style flowchart, tech-focused, neon highlights, showing data flow.
-        DATA PIPELINE
-        ├── INGESTION [Database Icon ~50px]
-        │   ├── STREAMING [Kafka Symbol]
-        │   └── BATCH [CSV/JSON Files]
-        ├── TRANSFORMATION [Gear Icon ~45px]
-        │   ├── CLEANING [Brush Icon]
-        │   └── AGGREGATION [Bar Graph]
-        ├── STORAGE [Cloud Icon ~50px]
-        └── ANALYTICS
-            └── DASHBOARDS [Monitor Icon]""",
+        "title": "E-Commerce Trends",
+        "prompt": """A hand-drawn infographic explaining e-commerce market trends:
+        - Title: "E-Commerce Trends 2025"
+        - Growth chart with arrow
+        - 4 key metrics: Conversion Rate, AOV, Repeat Customers, Mobile Share
+        - Minimal color scheme, doodle style icons""",
         "width": 1024,
         "height": 1024
     },
     {
-        "title": "Machine Learning Lifecycle",
-        "prompt": """A hand-drawn style flowchart, pastel palette, ML steps from data to deployment.
-        ML LIFECYCLE
-        ├── DATA COLLECTION [Folder Icon ~45px]
-        │   ├── DATA CLEANING [Soap Icon]
-        │   └── FEATURE ENGINEERING [Puzzle Icon]
-        ├── MODEL TRAINING [Robot Icon ~50px]
-        │   ├── HYPERPARAM TUNING [Dial Knob]
-        │   └── EVALUATION [Magnifier Icon]
-        ├── DEPLOYMENT [Cloud Icon ~45px]
-        └── MONITORING
-            └── FEEDBACK LOOP [Arrow Circle Icon]""",
+        "title": "Planet Sustainability",
+        "prompt": """A vivid infographic about environmental sustainability:
+        - Title: "Planet Sustainability Goals"
+        - Tree diagram with branches: Reduce, Reuse, Recycle
+        - Data highlights (CO2 reduction, forest coverage, plastic usage)
+        - Green and earthy color scheme""",
         "width": 1024,
         "height": 1024
     }
@@ -117,11 +90,11 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
         height=height,
         num_inference_steps=num_inference_steps,
         generator=generator,
-        guidance_scale=0.0  # 플로우차트 텍스트에 집중하되, 자유로운 표현
+        guidance_scale=0.0  # 인포그래픽 텍스트, 레이아웃에 집중
     ).images[0]
     return image, seed
 
-# CSS 스타일 (기존 구조 유지, 명칭만 일부 수정)
+# CSS 스타일 (기존 구조 유지)
 css = """
 .container {
     display: flex;
@@ -194,12 +167,11 @@ css = """
 }
 """
 
-# Gradio 인터페이스
 with gr.Blocks(css=css) as demo:
     gr.Markdown(
         """
-        <div class="title">GINI Flowchart</div>
-        <div class="subtitle">Create professional process flowcharts using FLUX AI</div>
+        <div class="title">GINI Infographic</div>
+        <div class="subtitle">Design stunning hand-drawn or modern style infographics with FLUX AI</div>
         """)
 
     with gr.Row(equal_height=True):
@@ -207,13 +179,13 @@ with gr.Blocks(css=css) as demo:
         with gr.Column(elem_id="input-column", scale=2):
             with gr.Group(elem_classes="input-box"):
                 prompt = gr.Text(
-                    label="Flowchart Prompt",
-                    placeholder="Enter your process flowchart structure...",
+                    label="Infographic Prompt",
+                    placeholder="Enter your infographic details...",
                     lines=10,
                     elem_classes="prompt-input"
                 )
-                run_button = gr.Button("Generate Flowchart", variant="primary")
-                result = gr.Image(label="Generated Flowchart")
+                run_button = gr.Button("Generate Infographic", variant="primary")
+                result = gr.Image(label="Generated Infographic")
             
             with gr.Accordion("Advanced Settings", open=False):
                 seed = gr.Slider(
@@ -251,7 +223,7 @@ with gr.Blocks(css=css) as demo:
         
         # 오른쪽 예제 컬럼
         with gr.Column(elem_id="examples-column", scale=1):
-            gr.Markdown("### Example Flowcharts")
+            gr.Markdown("### Example Infographics")
             for example in EXAMPLES:
                 with gr.Group(elem_classes="example-card"):
                     gr.Markdown(f"#### {example['title']}")