Update app.py

app.py

@@ -66,59 +66,141 @@ def update_random_seed():
     client = Client("http://211.233.58.201:7971/")
     return client.predict(api_name="/update_random_seed")
 
-# ===============================
-# UI 타이틀 및 설명
-# ===============================
-_TITLE = '''3D Llama'''
+# (이전 import 구문들은 동일)
+
+_TITLE = '''✨ 3D Llama Studio'''
 _DESCRIPTION = '''
-Text와 이미지를 이용하여 3D 모델을 생성할 수 있습니다.
+### Welcome to 3D Llama Studio - Your Advanced 3D Generation Platform
 
+This platform offers two powerful features:
+1. **Text/Image to 3D**: Generate detailed 3D models from text descriptions or reference images
+2. **Text to Styled Image**: Create artistic images that can be used for 3D generation
+
+*Note: Both English and Korean prompts are supported (영어와 한글 프롬프트 모두 지원됩니다)*
 '''
 
+# Custom CSS for dark theme and enhanced UI
+custom_css = """
+.gradio-container {
+    background-color: #1a1a1a;
+    color: #ffffff;
+}
+.tabs {
+    background-color: #2d2d2d;
+    border-radius: 10px;
+    padding: 10px;
+    margin: 10px 0;
+}
+.input-box {
+    background-color: #2d2d2d;
+    border: 1px solid #404040;
+    border-radius: 8px;
+    padding: 15px;
+    margin: 10px 0;
+}
+.button-primary {
+    background-color: #2d5a9e !important;
+    border: none !important;
+    color: white !important;
+}
+.button-secondary {
+    background-color: #404040 !important;
+    border: none !important;
+}
+"""
+
 def launch():
-    # 3D 모델 초기화
     model_zoo.init_models()
     
-    # Gradio Blocks 생성 (두 탭 포함)
-    with gr.Blocks(title=_TITLE) as demo:
+    with gr.Blocks(title=_TITLE, css=custom_css, theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate", font=["Arial", "sans-serif"])) as demo:
         with gr.Row():
-            gr.Markdown('# ' + _TITLE)
+            gr.Markdown('# ' + _TITLE, elem_classes="main-title")
         gr.Markdown(_DESCRIPTION)
         
-        # 탭 생성: 기존의 Text-to-3D와 새로 추가한 Text-to-IMAGE
-        with gr.Tabs():
-
-                
-            with gr.Tab("Text to 3D Style IMAGE"):
-                # 이미지 생성을 위한 파라미터 입력 컴포넌트 구성
-                with gr.Row():
-                    height_slider = gr.Slider(label="Height", minimum=256, maximum=2048, step=1, value=1024)
-                    width_slider = gr.Slider(label="Width", minimum=256, maximum=2048, step=1, value=1024)
-                with gr.Row():
-                    steps_slider = gr.Slider(label="Inference Steps", minimum=1, maximum=100, step=1, value=8)
-                    scales_slider = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=10.0, step=0.1, value=3.5)
-                prompt_text = gr.Textbox(label="Image Description", placeholder="Enter prompt here", lines=2)
-                seed_number = gr.Number(label="Seed (optional, leave empty for random)", value=None)
+        with gr.Tabs() as tabs:
+            with gr.Tab("🎨 Text to Styled Image", elem_classes="tab"):
+                with gr.Group(elem_classes="input-box"):
+                    gr.Markdown("### Image Generation Settings")
+                    with gr.Row():
+                        with gr.Column():
+                            height_slider = gr.Slider(
+                                label="Image Height",
+                                minimum=256,
+                                maximum=2048,
+                                step=64,
+                                value=1024,
+                                info="Select image height (pixels)"
+                            )
+                            width_slider = gr.Slider(
+                                label="Image Width",
+                                minimum=256,
+                                maximum=2048,
+                                step=64,
+                                value=1024,
+                                info="Select image width (pixels)"
+                            )
+                        with gr.Column():
+                            steps_slider = gr.Slider(
+                                label="Generation Steps",
+                                minimum=1,
+                                maximum=100,
+                                step=1,
+                                value=8,
+                                info="More steps = higher quality but slower"
+                            )
+                            scales_slider = gr.Slider(
+                                label="Guidance Scale",
+                                minimum=1.0,
+                                maximum=10.0,
+                                step=0.1,
+                                value=3.5,
+                                info="How closely to follow the prompt"
+                            )
+                    
+                    prompt_text = gr.Textbox(
+                        label="Image Description",
+                        placeholder="Enter your prompt here (English or Korean)",
+                        lines=3,
+                        elem_classes="input-box"
+                    )
+                    
+                    with gr.Row():
+                        seed_number = gr.Number(
+                            label="Seed",
+                            value=None,
+                            placeholder="Random seed if empty",
+                            elem_classes="input-box"
+                        )
+                        update_seed_button = gr.Button(
+                            "🎲 Random Seed",
+                            elem_classes="button-secondary"
+                        )
+                    
+                    generate_button = gr.Button(
+                        "🚀 Generate Image",
+                        elem_classes="button-primary"
+                    )
                 
-                # 'Update Random Seed' 버튼을 누르면 API를 통해 새로운 시드값을 받아 입력란 업데이트
-                update_seed_button = gr.Button("Update Random Seed")
-                update_seed_button.click(fn=update_random_seed, inputs=[], outputs=seed_number)
+                with gr.Group(elem_classes="input-box"):
+                    gr.Markdown("### Generated Result")
+                    image_output = gr.Image(label="Output Image")
                 
-                generate_button = gr.Button("Generate Image")
-                image_output = gr.Image(label="Generated Image")
+                update_seed_button.click(
+                    fn=update_random_seed,
+                    inputs=[],
+                    outputs=seed_number
+                )
                 
-                # 'Generate Image' 버튼 클릭 시 text_to_image 함수를 호출하여 결과 이미지를 출력
                 generate_button.click(
                     fn=text_to_image,
                     inputs=[height_slider, width_slider, steps_slider, scales_slider, prompt_text, seed_number],
                     outputs=image_output
                 )
 
-            with gr.Tab("Image to 3D"):
+            with gr.Tab("🎯 Image to 3D", elem_classes="tab"):
                 create_3d_ui("wkl")
                 
-    # 공유 링크를 생성하기 위해 share=True 설정
     demo.queue().launch(share=True)
-    
+
 if __name__ == '__main__':
-    fire.Fire(launch)
+    fire.Fire(launch)