Update app.py

app.py

@@ -35,6 +35,11 @@ torch.backends.cuda.matmul.allow_tf32 = True
 if not path.exists(gallery_path):
     os.makedirs(gallery_path, exist_ok=True)
 
+# 샘플 이미지와 프롬프트 정의
+SAMPLE_IMAGES = {
+    "3d1.webp": "the most famous hero according to Zhou Qi"
+}
+
 class timer:
     def __init__(self, method_name="timed process"):
         self.method = method_name
@@ -102,7 +107,7 @@ def save_image(image):
 examples = [
     ["A 3D Star Wars Darth Vader helmet, highly detailed metallic finish"],
     ["A 3D Iron Man mask with glowing eyes and metallic red-gold finish"],
-    ["A detailed 3D Pokemon Pikachu figure with glossy surface"],
+    ["A 3D Pokemon Pikachu figure with glossy surface"],
     ["A 3D geometric abstract cube transforming into a sphere, metallic finish"],
     ["A 3D steampunk mechanical heart with brass and copper details"],
     ["A 3D crystal dragon with transparent iridescent scales"],
@@ -154,7 +159,6 @@ def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt=
 def get_random_seed():
     return torch.randint(0, 1000000, (1,)).item()
 
-    
 def process_example(prompt):
     return process_and_save_image(
         height=1024,
@@ -165,7 +169,6 @@ def process_example(prompt):
         seed=get_random_seed()
     )
 
-    
 # Gradio 인터페이스
 with gr.Blocks(
     theme=gr.themes.Soft(),
@@ -192,83 +195,117 @@ with gr.Blocks(
         }
     """
 ) as demo:
-    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.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>
+                """
             )
             
-            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(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():
-                    steps = gr.Slider(
-                        label="Inference Steps",
-                        minimum=6,
-                        maximum=25,
-                        step=1,
-                        value=8
+                    
+                    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"]
                     )
-                    scales = gr.Slider(
-                        label="Guidance Scale",
-                        minimum=0.0,
-                        maximum=5.0,
-                        step=0.1,
-                        value=3.5
+
+                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.Tab("Gallery"):
+            with gr.Row():
+                gallery_html = "<div style='display: flex; flex-wrap: wrap; gap: 20px; padding: 20px;'>"
                 
-                seed = gr.Number(
-                    label="Seed (random by default, set for reproducibility)",
-                    value=get_random_seed(),
-                    precision=0
-                )
+                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>
+                        """
                 
-                randomize_seed = gr.Button("🎲 Randomize Seed", elem_classes=["generate-btn"])
-            
-            generate_btn = gr.Button(
-                "✨ Generate Image",
-                elem_classes=["generate-btn"]
-            )
-
-        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_html += "</div>"
+                gr.HTML(gallery_html)
 
     # Examples 섹션
     gr.Examples(
         examples=examples,
         inputs=prompt,
         outputs=output,
-        fn=process_example,  # 수정된 함수 사용
+        fn=process_example,
         cache_examples=False,
         examples_per_page=5
     )