Update app.py

app.py

@@ -1,25 +1,56 @@
 import gradio as gr
+import requests
+import os
 
-# 猜測晚餐的提示內容
-def guess_dinner(hint):
-    dinner_options = {
-        "It is cheesy and often served in slices.": "Pizza",
-        "It is a type of pasta usually served with a creamy sauce.": "Spaghetti",
-        "It is grilled, and some people like to add barbecue sauce.": "Grilled Chicken",
-        "It is usually found in a bun with lettuce and tomato.": "Hamburger",
-        "It is a popular dish in Asia, often served with soy sauce.": "Sushi"
+# 使用 Stable Diffusion 來生成晚餐圖片
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
+HF_API_TOKEN = os.getenv("HF_API_TOKEN")  # 確保您已在 Hugging Face Spaces 的 Secret 中設定 API token
+headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    
+    if response.status_code != 200:
+        print(f"Request failed with status code {response.status_code}")
+        print(f"Error details: {response.text}")
+        return f"Error: {response.status_code}, {response.text}"
+    
+    return response.content  # 返回圖片的二進位數據
+
+def generate_dinner_image(hint):
+    # 使用繁體中文的提示內容來對應不同的晚餐圖片描述
+    dinner_hints = {
+        "這是一個有很多起司的東西，通常切成片狀。": "a pizza with lots of cheese",
+        "這是一道義大利麵，通常搭配奶油醬。": "a plate of creamy spaghetti",
+        "這是一個烤的東西，有些人喜歡加上烤肉醬。": "grilled chicken with barbecue sauce",
+        "這通常在一個麵包中，裡面有生菜和番茄。": "a hamburger with lettuce and tomato",
+        "這是亞洲常見的食物，通常搭配醬油。": "sushi with soy sauce"
     }
     
-    # 根據提示猜測晚餐
-    return dinner_options.get(hint, "I don't know, please give me a valid hint!")
+    # 根據提示生成晚餐描述
+    description = dinner_hints.get(hint, "delicious food")
+    
+    # 調用 API 生成圖像
+    output = query({"inputs": description})
+    
+    # 如果返回錯誤信息，顯示錯誤
+    if isinstance(output, str) and output.startswith("Error:"):
+        return output  # 直接返回錯誤訊息
+    
+    # 保存生成的圖像到文件
+    with open("generated_dinner.png", "wb") as f:
+        f.write(output)
+    
+    # 返回生成的圖像文件
+    return "generated_dinner.png"
 
 # 創建 Gradio 介面
 interface = gr.Interface(
-    fn=guess_dinner,
-    inputs=gr.Textbox(lines=2, placeholder="輸入提示內容來猜猜我的晚餐..."),
-    outputs="text",
+    fn=generate_dinner_image,
+    inputs=gr.Textbox(lines=2, placeholder="輸入提示內容來生成晚餐圖片...（例如：這是一個有很多起司的東西，通常切成片狀。）"),
+    outputs="image",  # 生成圖片
     title="最強第一組",
-    description="給我提示內容，讓我猜猜你的晚餐!"
+    description="根據您的描述生成晚餐圖片！輸入提示內容（例如：這是一個有很多起司的東西，通常切成片狀）"
 )
 
 # 啟動 Gradio 應用