Update app.py

app.py

@@ -14,19 +14,25 @@ def set_header():
     }
     return header
 
-# 비디오 또는 오디오 컨텐츠 생성하는 함수.
-def generate_contents(user_content, brand_name):
+# 비디오 또는 오디오 컨텐츠 생성하는 함수
+def generate_contents(user_content, brand_name, aspect_ratio='landscape'):
     url = API_HOME + END_GENERATE
-    VOICE_ID = '65934ac2bc02ab1c006755fa'  # Korean, Korea, Wolf Sea
+    VOICE_ID = '64ea14306a00ac991dcb8a3f'  # Korean, Korea, Wolf Sea
 
-    # 각 문장을 별도의 scene으로 처리
-    scenes = [{"content": line, "voiceId": VOICE_ID} for line in user_content.split('\n') if line.strip()]
+    # 멀티 라인 텍스트를 분리하여 각 줄을 별도의 씬으로 처리
+    scenes = [{"content": line.strip(), "voiceId": VOICE_ID} for line in user_content.split('\n') if line.strip()]
 
     payload = {
         "format": "video",  # video | audio
         "scenes": scenes,
         "settings": {
-            # 설정들...
+            'aspectRatio': aspect_ratio,
+            'subtitle': {
+                'fontColor': 'yellow',
+                'backgroundColor': 'black',
+                'placement': 'bottom',
+                'display': 'phrase',
+            },
         },
         "backgroundMusicKeywords": "happy, lofi, beats"
     }
@@ -39,7 +45,7 @@ def generate_contents(user_content, brand_name):
     else:
         return 'FAILED'
 
-# 컨텐츠 생성 상태를 확인하는 함수.
+# 컨텐츠 생성 상태를 확인하는 함수
 def generate_status(content_id):
     url = API_HOME + END_STATUS
     payload = {"id": content_id}
@@ -59,6 +65,9 @@ def generate_status(content_id):
     return 'FAILED'
 
 # Gradio 인터페이스 함수
+def gradio_generate_contents(user_content, brand_name, aspect_ratio):
+    return generate_contents(user_content, brand_name, aspect_ratio)
+
 def gradio_generate_status(content_id):
     download_url = generate_status(content_id)
     if download_url != 'FAILED':
@@ -69,10 +78,24 @@ def gradio_generate_status(content_id):
     else:
         return "컨텐츠 생성 실패 또는 ID가 잘못되었습니다."
 
-status_iface = gr.Interface(
+# 컨텐츠 생성 인터페이스
+iface_create = gr.Interface(
+    fn=gradio_generate_contents,
+    inputs=[
+        gr.TextArea(label="컨텐츠 텍스트"),
+        gr.Textbox(label="브랜드명"),
+        gr.Dropdown(label="화면 비율", choices=['portrait', 'square', 'landscape'], value='landscape')
+    ],
+    outputs=gr.Textbox(label="컨텐츠 ID")
+)
+
+# 상태 확인 및 비디오 다운로드 인터페이스
+iface_status = gr.Interface(
     fn=gradio_generate_status,
     inputs=gr.Textbox(label="컨텐츠 ID"),
     outputs=gr.HTML(label="다운로드 링크 및 비디오")
 )
 
-status_iface.launch()
+# 두 인터페이스를 탭으로 구성
+iface_combined = gr.TabbedInterface([iface_create, iface_status], ["컨텐츠 생성", "상태 확인 및 다운로드"])
+iface_combined.launch()