Update app.py

app.py

@@ -5,27 +5,34 @@ API_KEY = "ZZUIQ4OZASNRQ8B8WYHNW"
 API_URL = "https://api.fliki.ai/v1/generate/text-to-image"
 
 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)
+    headers = {
+        'Authorization': f'Bearer {API_KEY}',
+        'Content-Type': 'application/json',
+    }
+    json_data = {
+        'content': content,
+        'style': style,
+    }
+    response = requests.post(API_URL, headers=headers, json=json_data)
+
     if response.status_code == 200:
-        return response.json()['url']  # 이미지 다운로드 URL 반환
+        return response.json()['url']
     else:
         raise Exception(f"오류 발생: {response.status_code}")
 
+styles = [
+    "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"
+]
+
 with gr.Blocks() as demo:
     with gr.Row():
         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="스타일 선택"
-        )
+        style_input = gr.Dropdown(choices=styles, label="스타일 선택")
         submit_button = gr.Button("이미지 생성")
-    output = gr.Image(label="생성된 이미지")
+    output = gr.Image(label="생성된 이미지", type="pil")
 
     submit_button.click(
         generate_image,