firebase init

app.py

@@ -1,4 +1,7 @@
 import gradio as gr
+import firebase_admin
+from firebase_admin import credentials, storage
+import tempfile
 import os
 import io
 from PIL import Image
@@ -12,15 +15,53 @@ initialize(_use_local=False, use_gpu=True, use_dotenv=False)
 init_model(use_local=False)
 load_wd14_tagger_model()
 
+# Firebase の初期化
+cred = credentials.Certificate("firebase/image2body-demo-firebase-adminsdk-ope1k-e19e1da82c.json")
+firebase_admin.initialize_app(cred, {
+    'storageBucket': 'image2body-demo.appspot.com'
+})
+def save_image_pair_to_firebase(sotai_image_data, sketch_image_data):
+    # 一意の識別子を生成
+    unique_id = str(uuid.uuid4())
+    timestamp = int(time.time())
+    folder_name = f"image_pairs/{timestamp}_{unique_id}"
+
+    # Base64 データを PIL Image に変換
+    sotai_image = Image.open(io.BytesIO(base64.b64decode(sotai_image_data)))
+    sketch_image = Image.open(io.BytesIO(base64.b64decode(sketch_image_data)))
+    
+    bucket = storage.bucket()
+    urls = {}
+
+    for image_type, image in [("sotai", sotai_image), ("sketch", sketch_image)]:
+        # 一時ファイルを作成
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+            image.save(temp_file, format="PNG")
+            temp_file_path = temp_file.name
+
+        try:
+            # Firebase Storage にアップロード
+            blob_path = f'{folder_name}/{image_type}.png'
+            blob = bucket.blob(blob_path)
+            blob.upload_from_filename(temp_file_path)
+            
+            # URLを取得
+            blob.make_public()
+            urls[image_type] = blob.public_url
+        finally:
+            # 一時ファイルを削除
+            os.unlink(temp_file_path)
+    
+    return urls
+
 def process_image(input_image, mode, weight1, weight2):
-    # 画像処理ロジック
+    # 既存の画像処理ロジック
     sotai_image, sketch_image = process_image_as_base64(input_image, mode, weight1, weight2)
     
-    # Base64文字列をPIL Imageに変換
-    sotai_pil = Image.open(io.BytesIO(base64.b64decode(sotai_image)))
-    sketch_pil = Image.open(io.BytesIO(base64.b64decode(sketch_image)))
+    # Firebase に画像ペアを保存し、URLを取得
+    urls = save_image_pair_to_firebase(sotai_image, sketch_image)
     
-    return sotai_pil, sketch_pil
+    return urls['sotai'], urls['sketch']
 
 # Gradio インターフェースの定義
 iface = gr.Interface(
@@ -32,8 +73,8 @@ iface = gr.Interface(
         gr.Slider(0, 1, value=0.05, step=0.025, label="Weight 2 (Body)")
     ],
     outputs=[
-        gr.Image(type="pil", label="Sotai (Body) Image"),
-        gr.Image(type="pil", label="Sketch Image")
+        gr.Text(label="Sotai Image URL"),
+        gr.Text(label="Sketch Image URL")
     ],
     title="Image2Body API",
     description="Upload an image and select processing options to generate body and sketch images."

firebase/image2body-demo-firebase-adminsdk-ope1k-e19e1da82c.json

@@ -0,0 +1,13 @@
+{
+  "type": "service_account",
+  "project_id": "image2body-demo",
+  "private_key_id": "e19e1da82c39206ae07a42271eb6e9a067b501b6",
+  "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQDp/N0pO3GUp5/k\nKvzNRBrmN8EgPwqDT+SeQYWn8po+xmtWUcrLlUH1ZgRTaJvkdgxxSsGfoOSj1mjw\nVSYudfT7763PTznqHWi7ShHvLwriuoEaJkwN4yhLyayTR42AQelPJUkl0FvE/DO7\neVjYtbReXH5+lSFjTYK7/Z0mS1D5Bc3ubEUDjJb0/ySc+i4Q/hmy+RiwDwkpTDPu\nufMUgcGHvv1ApMiwqXQMv2OuBigW1XS+IpNhwmTP5c7DlwnFm8qaVnwQcaj9NbPW\nNP2wrEjkzo2piY33GZ+dTV7wwAdJLjY30dKcEdkrcRU0dhDOZ++hwhGB4hecGnlR\nHbrS58AbAgMBAAECggEARC8g3WaKkwiDsdALCd+6PPLrTrPAIwKxeqJaSdYKt5dH\naQh9oXSVm2aGhS27G8iAmLTxlkwixFFzE/GkJi7OXGn8CqiqFjxKrCUJ89qQAADc\nJRwa1oMm12rl5KcZp1HNpHgXiDLewxjBFHTofIH6lPK8E7PuXksUeSaO2D1jU6Uc\nOLJJuHHkZ52Fn34N3XjUwPJ6dOLVyImUY1IBJ7cdwRuvfohmcinAlvEqtnqA7hOa\n0QZdTcFpkIHq0yahqMYulpLCG6B8hNaRcHeET+Kl14cSn5Y5ifmGKe8haucGN/Zb\nwA1RO/RaRbbYF93nL3KENpegtb0rMej2Acvl7q0nzQKBgQD1UbxgZaDrUPEIZZDs\nCmK8GLDDqUuVpVxPUi0aMoI0zE7Li1p9OFWSySvqeAmJnMRRGrbN46WfdbkRYIlD\no6+Gd5E0SJAsDwZqyx7Apg2Jrx7AdSXAdyZn83F3R6yi+L63zcr3ND6KDUXMCtwa\no7ISgdlR1RxG1K/9FXisSosdpwKBgQD0LNRt34k6GDWjWCS/VZzbdR7+I0Bt9D0K\nCFqSjRMYL9+93DFQqCkN+ueZvGUQhJUm7XJ0wpF7qO97TnwNR784iYO28UKtofvA\n0Mog31YTb+Dm1SnbvxzzpBPaLLgkommBc5qKqgtg8gniJJ8ddWZYAYCWsPMxJwSA\nf74uYbbgbQKBgQDZJOJTiWGpKVpRDmX7KS5ARLKiXoYzDc3pwFl/6NlsFrRgYxdT\n47uLVFp1Do+SqZr0F2tcT3bY8QFBqmuV3ZuOsONsnhZPiXUqrmSoLdOp3cKgpJEo\nDc+HD+uNo3niP7/BPn271xjMZfc0JyCwy9jhnOJaFwMSh0zH/mt0Rf9h3QKBgQC5\nMU8kzkfcQgRckTu9yocfg6MwCTiSrI2nEFLxmAIFdDn0XA8jJ4eLzOLfnR3h+0zM\nq4cFu5Rv/98y3IIF6IbFsLaviCiUXKj+wtyrNfio4U/og4At5LNmrulHrmczkeMH\nhdCOnw3mR2zrke8EfshSs0D+CdNJUlh9xWVO9FmpuQKBgAqCGUXvTKKrSTmQS3Cv\nwvndgkBrl21zeOkIpDKFdHhMYgVFSF5CS3FIoSnbZrakHrEDYItG+UjVepH9kvmn\nk4KPZJ4CzO3qwnKIdc28kSfT1tK3GDC3aufh9ucWuGSfHO7j6D7yM73Lb/FsRpju\nA8nojFAbWcfdKpCjh1xiRDTF\n-----END PRIVATE KEY-----\n",
+  "client_email": "firebase-adminsdk-ope1k@image2body-demo.iam.gserviceaccount.com",
+  "client_id": "105880637239614588319",
+  "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+  "token_uri": "https://oauth2.googleapis.com/token",
+  "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+  "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/firebase-adminsdk-ope1k%40image2body-demo.iam.gserviceaccount.com",
+  "universe_domain": "googleapis.com"
+}

requirements.txt

@@ -22,4 +22,5 @@ python-dotenv
 peft==0.11.1
 flask_limiter==3.7.0
 gevent==24.2.1
-gevent-websocket==0.10.1
+gevent-websocket==0.10.1
+firebase-admin