Update app.py

app.py

@@ -1,44 +1,30 @@
 import gradio as gr
 import requests
 
-# API 키 설정
 API_KEY = "ZZUIQ4OZASNRQ8B8WYHNW"
 
-def generate_video(keyword, voice_actor, content_type, style):
-    # Fliki API에 비디오 생성 요청을 보내는 코드
-    # 이 부분은 Fliki API 문서에 따라 적절하게 구현해야 합니다.
-    # 예시:
-    url = "https://api.fliki.ai/v1/generate/video"
-    payload = {
-        "keyword": keyword,
-        "voice_actor": voice_actor,
-        "content_type": content_type,
-        "style": style
-    }
+def generate_content(text, content_type):
+    url = f"https://api.fliki.ai/v1/generate/{content_type}"
+    payload = {"text": text}
     headers = {"Authorization": f"Bearer {API_KEY}"}
+
     response = requests.post(url, json=payload, headers=headers)
-    
     if response.status_code == 200:
-        return response.json()["video_url"]  # 비디오 URL 반환
+        return response.json()['url']  # 비디오 또는 오디오 URL 반환
     else:
-        raise Exception(f"오류 발생. 상태 코드: {response.status_code}")
+        raise Exception(f"오류 발생: {response.status_code}")
 
-# Gradio 인터페이스 정의
 with gr.Blocks() as demo:
     with gr.Row():
-        keyword_input = gr.Textbox(label="키워드 입력")
-        voice_actor_input = gr.Dropdown(choices=["성우1", "성우2", "성우3"], label="성우 선택")
-        content_type_input = gr.Radio(choices=["이미지", "비디오"], label="콘텐츠 유형")
-        style_input = gr.Dropdown(choices=["스타일1", "스타일2", "스타일3"], label="스타일 선택")
+        text_input = gr.Textbox(label="텍스트 입력")
+        content_type_input = gr.Dropdown(choices=["text-to-speech", "text-to-video"], label="콘텐츠 유형")
         submit_button = gr.Button("생성")
-    
-    video_output = gr.Video(label="생성된 비디오")
+    output = gr.Video(label="생성된 콘텐츠")
 
     submit_button.click(
-        fn=generate_video,
-        inputs=[keyword_input, voice_actor_input, content_type_input, style_input],
-        outputs=video_output
+        generate_content,
+        inputs=[text_input, content_type_input],
+        outputs=output
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()