Update app.py

app.py

@@ -170,7 +170,7 @@ def process_example(prompt):
         seed=get_random_seed()
     )
 
-# Gradio 인터페이스
+# Gradio 인터페이스 부분을 다음과 같이 수정
 with gr.Blocks(
     theme=gr.themes.Soft(),
     css="""
@@ -194,116 +194,130 @@ with gr.Blocks(
             max-width: 1024px;
             margin: auto;
         }
+        .gallery-container {
+            margin-top: 40px;
+            padding: 20px;
+            background: #f5f5f5;
+            border-radius: 15px;
+        }
+        .gallery-title {
+            text-align: center;
+            margin-bottom: 20px;
+            color: #333;
+            font-size: 1.5rem;
+        }
     """
 ) as demo:
-    with gr.Tabs():
-        # 첫 번째 탭 - Generator
-        with gr.Tab("Generator"):
-            gr.HTML(
-                """
-                <div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
-                    <h1 style="font-size: 2.5rem; color: #2196F3;">3D Style Image Generator</h1>
-                    <p style="font-size: 1.2rem; color: #666;">Create amazing 3D-style images with AI</p>
-                </div>
-                """
+    gr.HTML(
+        """
+        <div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
+            <h1 style="font-size: 2.5rem; color: #2196F3;">3D Style Image Generator</h1>
+            <p style="font-size: 1.2rem; color: #666;">Create amazing 3D-style images with AI</p>
+        </div>
+        """
+    )
+    
+    with gr.Row(elem_classes="container"):
+        with gr.Column(scale=3):
+            prompt = gr.Textbox(
+                label="Image Description",
+                placeholder="Describe the 3D image you want to create...",
+                lines=3
             )
             
-            with gr.Row(elem_classes="container"):
-                with gr.Column(scale=3):
-                    prompt = gr.Textbox(
-                        label="Image Description",
-                        placeholder="Describe the 3D image you want to create...",
-                        lines=3
+            with gr.Accordion("Advanced Settings", open=False):
+                with gr.Row():
+                    height = gr.Slider(
+                        label="Height",
+                        minimum=256,
+                        maximum=1152,
+                        step=64,
+                        value=1024
                     )
-                    
-                    with gr.Accordion("Advanced Settings", open=False):
-                        with gr.Row():
-                            height = gr.Slider(
-                                label="Height",
-                                minimum=256,
-                                maximum=1152,
-                                step=64,
-                                value=1024
-                            )
-                            width = gr.Slider(
-                                label="Width",
-                                minimum=256,
-                                maximum=1152,
-                                step=64,
-                                value=1024
-                            )
-                        
-                        with gr.Row():
-                            steps = gr.Slider(
-                                label="Inference Steps",
-                                minimum=6,
-                                maximum=25,
-                                step=1,
-                                value=8
-                            )
-                            scales = gr.Slider(
-                                label="Guidance Scale",
-                                minimum=0.0,
-                                maximum=5.0,
-                                step=0.1,
-                                value=3.5
-                            )
-                        
-                        seed = gr.Number(
-                            label="Seed (random by default, set for reproducibility)",
-                            value=get_random_seed(),
-                            precision=0
-                        )
-                        
-                        randomize_seed = gr.Button("🎲 Randomize Seed", elem_classes=["generate-btn"])
-                    
-                    generate_btn = gr.Button(
-                        "✨ Generate Image",
-                        elem_classes=["generate-btn"]
+                    width = gr.Slider(
+                        label="Width",
+                        minimum=256,
+                        maximum=1152,
+                        step=64,
+                        value=1024
                     )
-
-                with gr.Column(scale=4, elem_classes=["fixed-width"]):
-                    output = gr.Image(
-                        label="Generated Image",
-                        elem_id="output-image",
-                        elem_classes=["output-image", "fixed-width"],
-                        value="3d.webp"
+                
+                with gr.Row():
+                    steps = gr.Slider(
+                        label="Inference Steps",
+                        minimum=6,
+                        maximum=25,
+                        step=1,
+                        value=8
+                    )
+                    scales = gr.Slider(
+                        label="Guidance Scale",
+                        minimum=0.0,
+                        maximum=5.0,
+                        step=0.1,
+                        value=3.5
                     )
-
-        # 두 번째 탭 - Gallery
-        with gr.Tab("Gallery"):
-            with gr.Row():
-                gallery_html = "<div style='display: flex; flex-wrap: wrap; gap: 20px; padding: 20px;'>"
                 
-                for img_file, prompt in SAMPLE_IMAGES.items():
-                    if os.path.exists(img_file):
-                        gallery_html += f"""
-                        <div style='
-                            border: 1px solid #ddd;
-                            border-radius: 10px;
-                            padding: 10px;
-                            width: 300px;
-                            box-shadow: 0 4px 8px rgba(0,0,0,0.1);
-                        '>
-                            <img src='file/{img_file}' 
-                                style='width: 100%; 
-                                border-radius: 8px;
-                                margin-bottom: 10px;'
-                            >
-                            <p style='
-                                margin: 5px 0;
-                                font-weight: bold;
-                                color: #333;
-                            '>Prompt: {prompt}</p>
-                        </div>
-                        """
+                seed = gr.Number(
+                    label="Seed (random by default, set for reproducibility)",
+                    value=get_random_seed(),
+                    precision=0
+                )
                 
-                gallery_html += "</div>"
-                gr.HTML(gallery_html)
+                randomize_seed = gr.Button("🎲 Randomize Seed", elem_classes=["generate-btn"])
+            
+            generate_btn = gr.Button(
+                "✨ Generate Image",
+                elem_classes=["generate-btn"]
+            )
 
-# Gradio 인터페이스 부분에서 Examples 섹션을 다음과 같이 수정
+        with gr.Column(scale=4, elem_classes=["fixed-width"]):
+            output = gr.Image(
+                label="Generated Image",
+                elem_id="output-image",
+                elem_classes=["output-image", "fixed-width"],
+                value="3d.webp"
+            )
 
+    # Gallery 섹션
+    with gr.Row(elem_classes="gallery-container"):
+        gr.HTML("<h2 class='gallery-title'>Gallery</h2>")
+        gallery_html = "<div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>"
+        
+        for img_file, prompt in SAMPLE_IMAGES.items():
+            if os.path.exists(img_file):
+                gallery_html += f"""
+                <div style='
+                    border: 1px solid #ddd;
+                    border-radius: 10px;
+                    padding: 10px;
+                    background: white;
+                    box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+                '>
+                    <img src='file/{img_file}' 
+                        style='width: 100%; 
+                        border-radius: 8px;
+                        margin-bottom: 10px;'
+                    >
+                    <p style='
+                        margin: 5px 0;
+                        font-weight: bold;
+                        color: #333;
+                        padding: 10px;
+                    '>Prompt: {prompt}</p>
+                </div>
+                """
+        
+        gallery_html += "</div>"
+        gr.HTML(gallery_html)
 
+    # Examples 섹션
+    with gr.Row():
+        gr.Examples(
+            label="Example Prompts",
+            examples=example_prompts,
+            inputs=prompt
+        )
 
     def update_seed():
         return get_random_seed()