Update app.py

app.py

@@ -1,36 +1,36 @@
 import gradio as gr
 import requests
+from PIL import Image
+import io
 
 API_KEY = "ZZUIQ4OZASNRQ8B8WYHNW"
-API_URL = "https://api.fliki.ai/v1/generate/text-to-image"
 
 def generate_image(content, style):
     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)
-
+    response = requests.post('https://api.fliki.ai/v1/generate/text-to-image', headers=headers, json=json_data)
+    
     if response.status_code == 200:
-        return response.json()['url']
+        image_url = response.json()['url']
+        image_response = requests.get(image_url)
+        image = Image.open(io.BytesIO(image_response.content))
+        return image
     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"
-]
+        raise Exception(f"오류 발생: {response.text}")
 
 with gr.Blocks() as demo:
     with gr.Row():
         content_input = gr.Textbox(label="텍스트 입력")
-        style_input = gr.Dropdown(choices=styles, 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="생성된 이미지", type="pil")