Update app.py

app.py

@@ -1,38 +1,37 @@
-import gradio as gr
 import requests
-import os
+import base64
 import json
+import os
 
-NANO_BANANA_API_KEY = os.getenv("NANO_BANANA_API_KEY")
-
-# --- ここに関数追加 ---
-def segment_image(image):
-    headers = {"Authorization": f"Bearer {NANO_BANANA_API_KEY}"}
-    files = {"file": open(image, "rb")}
-    response = requests.post(
-        "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent", files=files, headers=headers
-    )
-
+API_KEY = os.getenv("NANO_NABABA_API_KEY")  # Hugging Face Secrets に登録したキー
+API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro:generateContent"
+
+def generate_completion(image_path, prompt="欠けている部分を補完してください。"):
+    with open(image_path, "rb") as f:
+        image_data = base64.b64encode(f.read()).decode("utf-8")
+
+    headers = {
+        "Content-Type": "application/json",
+        "x-goog-api-key": API_KEY
+    }
+
+    payload = {
+        "contents": [
+            {
+                "parts": [
+                    {"text": prompt},
+                    {
+                        "inline_data": {
+                            "mime_type": "image/png",
+                            "data": image_data
+                        }
+                    }
+                ]
+            }
+        ]
+    }
+
+    response = requests.post(API_URL, headers=headers, json=payload)
     print("Response status:", response.status_code)
-    try:
-        print(json.dumps(response.json(), indent=2, ensure_ascii=False))
-    except Exception as e:
-        print("JSON parse error:", e)
-
-    if response.status_code == 200:
-        data = response.json()
-        if "images" in data:
-            return [img["url"] for img in data["images"]]
-    return ["処理に失敗しました"]
-
-# --- Gradio UI 部分 ---
-with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 Nano Banana 画像分割デモ")
-
-    input_image = gr.Image(label="入力画像", type="filepath")
-    run_btn = gr.Button("分割実行")
-    output_gallery = gr.Gallery(label="分割結果", show_label=True, columns=2, height="auto")
-
-    run_btn.click(segment_image, inputs=[input_image], outputs=[output_gallery])
-
-demo.launch()
+    print(response.text)
+    return response