Update app.py

app.py

@@ -51,9 +51,11 @@ def text_to_image(height, width, steps, scales, prompt, seed):
         seed,
         api_name="/process_and_save_image"
     )
-    # API 응답은 dict 형태이며, "url" 혹은 "path"에 생성된 이미지 정보가 있다.
-    # 여기서는 "url" 값을 반환하도록 한다.
-    return result.get("url", None)
+    # 결과가 dict이면 "url" 키의 값을, 아니라면 그대로 반환합니다.
+    if isinstance(result, dict):
+        return result.get("url", None)
+    else:
+        return result
 
 def update_random_seed():
     """
@@ -67,10 +69,10 @@ def update_random_seed():
 # ===============================
 # UI 타이틀 및 설명
 # ===============================
-_TITLE = '''Text to 3D & Image Generation'''
+_TITLE = '''3D-llama'''
 _DESCRIPTION = '''
-Text를 이용하여 3D 모델과 이미지를 생성할 수 있습니다.
-왼쪽 탭은 3D 모델 생성을, 오른쪽 탭은 이미지 생성을 담당합니다.
+Text와 이미지를 이용하여 3D 모델을 생성할 수 있습니다.
+
 '''
 
 def launch():
@@ -85,10 +87,9 @@ def launch():
         
         # 탭 생성: 기존의 Text-to-3D와 새로 추가한 Text-to-IMAGE
         with gr.Tabs():
-            with gr.Tab("Text to 3D"):
-                create_3d_ui("wkl")
+
                 
-            with gr.Tab("Text to IMAGE"):
+            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)
@@ -112,7 +113,11 @@ def launch():
                     inputs=[height_slider, width_slider, steps_slider, scales_slider, prompt_text, seed_number],
                     outputs=image_output
                 )
+
+            with gr.Tab("Image to 3D"):
+                create_3d_ui("wkl")
                 
+    # 공유 링크를 생성하기 위해 share=True 설정
     demo.queue().launch(share=True)
     
 if __name__ == '__main__':