Update app.py

app.py

@@ -2,33 +2,34 @@ import gradio as gr
 import requests
 
 API_KEY = "ZZUIQ4OZASNRQ8B8WYHNW"
-API_URL = "https://api.fliki.ai/v1"
+API_URL = "https://api.fliki.ai/v1/generate/text-to-image"
 
-def generate_content(text, format):
-    url = f"{API_URL}/generate"
-    payload = {
-        "format": format,
-        "scenes": [{"content": text}],
-        # 추가 설정이 필요하면 여기에 추가
-    }
-    headers = {"Authorization": f"Bearer {API_KEY}"}
-
-    response = requests.post(url, json=payload, headers=headers)
+def generate_image(content, style):
+    payload = {"content": content, "style": style}
+    headers = {"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"}
+    
+    response = requests.post(API_URL, json=payload, headers=headers)
     if response.status_code == 200:
-        return response.json()['url']  # 응답에서 URL 추출
+        return response.json()['url']  # 이미지 다운로드 URL 반환
     else:
         raise Exception(f"오류 발생: {response.status_code}")
 
 with gr.Blocks() as demo:
     with gr.Row():
-        text_input = gr.Textbox(label="텍스트 입력")
-        format_input = gr.Dropdown(choices=["text-to-speech", "text-to-video"], label="포맷 선택")
-        submit_button = gr.Button("생성")
-    output = gr.Video(label="생성된 콘텐츠")
+        content_input = gr.Textbox(label="텍스트 입력")
+        style_input = gr.Dropdown(
+            choices=["3d-model", "analog-film", "anime", "cinematic", "comic-book", 
+                     "digital-art", "enhance", "fantasy-art", "isometric", "line-art", 
+                     "low-poly", "modeling-compound", "neon-punk", "origami", 
+                     "photographic", "pixel-art", "tile-texture"], 
+            label="스타일 선택"
+        )
+        submit_button = gr.Button("이미지 생성")
+    output = gr.Image(label="생성된 이미지")
 
     submit_button.click(
-        generate_content,
-        inputs=[text_input, format_input],
+        generate_image,
+        inputs=[content_input, style_input],
         outputs=output
     )