1

app.py

@@ -1,7 +1,9 @@
 import requests
 import gradio as gr
+from PIL import Image
+import io
 
-# 修改 greet 函数以处理图像生成
+# 修改 generate_image 函数以处理图像生成并保存为临时文件
 def generate_image(api_key, prompt, aspect_ratio, model, negative_prompt, seed):
     response = requests.post(
         "https://api.stability.ai/v2beta/stable-image/generate/sd3",
@@ -21,7 +23,13 @@ def generate_image(api_key, prompt, aspect_ratio, model, negative_prompt, seed):
     )
     
     if response.status_code == 200:
-        return response.content
+        # 将 bytes 数据转换为 PIL Image 对象
+        image_bytes = io.BytesIO(response.content)
+        image = Image.open(image_bytes)
+        # 保存为临时文件
+        temp_path = "./temp_image.jpeg"
+        image.save(temp_path)
+        return temp_path
     else:
         raise Exception(str(response.json()))
 
@@ -43,4 +51,4 @@ with gr.Blocks() as demo:
                         inputs=[api_key, prompt, aspect_ratio, model, negative_prompt, seed], 
                         outputs=output)
 
-demo.launch(server_port=7588)
+demo.launch(server_port=7855)  

requirements.txt

@@ -1,2 +1,3 @@
 gradio
-requests
+requests
+pillow