Update app.py

app.py

@@ -1,64 +1,131 @@
 import gradio as gr
 import base64
-import os # ファイルパスのデバッグ用に念のためインポート
+import os
+import io
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import JSONResponse
+import uvicorn # ローカル実行用にインポート
 
-def encode_to_base64(file_obj):
+# --- Base64 エンコード処理 (共通ロジック) ---
+def encode_bytes_to_base64(file_bytes: bytes) -> str:
     """
-    gr.Fileから受け取ったファイルオブジェクトを読み込み、
-    Base64エンコードされた文字列を返す関数。
+    バイト列を受け取り、Base64エンコードされた文字列を返す。
+    エラーが発生した場合は例外を送出する。
+    """
+    try:
+        base64_bytes = base64.b64encode(file_bytes)
+        base64_string = base64_bytes.decode('utf-8')
+        return base64_string
+    except Exception as e:
+        print(f"Base64エンコードエラー: {e}")
+        # エラーを呼び出し元に伝えるために再度raiseする
+        raise ValueError(f"Base64エンコード中にエラーが発生しました: {e}")
+
+# --- Gradio用 関数 ---
+def gradio_encode_handler(file_obj):
+    """
+    GradioのFileコンポーネントから受け取ったファイルを処理する関数。
     """
     if file_obj is None:
         return "ファイルをアップロードしてください。"
 
     try:
-        # GradioのFileコンポーネントは一時ファイルオブジェクトを提供し、
-        # その .name 属性に一時ファイルのパスが含まれている
         file_path = file_obj.name
-        print(f"処理中のファイル: {file_path}") # デバッグ用ログ
+        print(f"Gradio処理中のファイル: {file_path}") # デバッグ用ログ
 
-        # ファイルをバイナリ読み込みモードで開く
         with open(file_path, 'rb') as f:
-            # ファイルの内容全体をバイト列として読み込む
             file_bytes = f.read()
 
-        # バイト列をBase64にエンコードする (結果もバイト列)
-        base64_bytes = base64.b64encode(file_bytes)
-
-        # Base64のバイト列をUTF-8文字列にデコードする
-        base64_string = base64_bytes.decode('utf-8')
-
+        # 共通のエンコード関数を呼び出す
+        base64_string = encode_bytes_to_base64(file_bytes)
         return base64_string
 
     except Exception as e:
-        print(f"エラー発生: {e}") # デバッグ用ログ
+        print(f"Gradioファイル処理エラー: {e}") # デバッグ用ログ
+        # Gradioではエラーメッセージを文字列として返す
         return f"ファイルの処理中にエラーが発生しました: {e}"
 
-# --- Gradioインターフェースの定義 ---
+# --- FastAPI アプリケーションの初期化 ---
+app = FastAPI(
+    title="File to Base64 API & Gradio UI",
+    description="ファイルを受け取りBase64にエンコードするAPIとGradio UIを提供します。"
+)
 
-# 入力コンポーネント: ファイルアップロード
-# file_typesで使用可能なファイル種別をユーザーに示す（画像全般とPDFを指定）
-input_file = gr.File(
-    label="画像またはPDFファイルを入力",
-    file_types=['image', '.pdf'] # 'image'は一般的な画像形式をカバー, '.pdf'でPDFを指定
+# --- FastAPI エンドポイントの定義 ---
+@app.post(
+    "/encode/",
+    summary="ファイルをBase64エンコード",
+    description="アップロードされたファイルを読み込み、Base64エンコードされた文字列をJSON形式で返します。",
+    response_description="ファイル名とBase64エンコードされた文字列を含むJSONオブジェクト"
 )
+async def api_encode_file(file: UploadFile = File(..., description="Base64エンコードする画像またはPDFファイル")):
+    """
+    FastAPIのエンドポイント。
+    POSTリクエストでファイルを受け取り、Base64エンコードしてJSONで返す。
+    """
+    print(f"API処理中のファイル: {file.filename}, Content-Type: {file.content_type}")
+
+    # ファイルの内容を非同期で読み込む
+    contents = await file.read()
+
+    # ファイルが空でないかチェック
+    if not contents:
+        raise HTTPException(status_code=400, detail="空のファイルは処理できません。")
 
-# 出力コンポーネント: テキストボックス
-# linesで高さを調整し、interactive=Trueでコピー可能にする
-output_text = gr.Textbox(
+    try:
+        # 共通のエンコード関数を呼び出す
+        base64_string = encode_bytes_to_base64(contents)
+        # 成功したらJSONレスポンスを返す
+        return JSONResponse(
+            content={
+                "filename": file.filename,
+                "content_type": file.content_type,
+                "base64_string": base64_string
+            }
+        )
+    except ValueError as e: # Base64エンコード中のエラーを捕捉
+        print(f"APIエンコード処理エラー: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+    except Exception as e: # その他の予期せぬエラー
+        print(f"API予期せぬエラー: {e}")
+        raise HTTPException(status_code=500, detail=f"ファイルの処理中に予期せぬエラーが発生しました: {e}")
+    finally:
+        # アップロードされたファイルを閉じる (重要)
+        await file.close()
+
+# --- Gradio インターフェースの定義 ---
+input_file_gr = gr.File(
+    label="画像またはPDFファイルを入力",
+    file_types=['image', '.pdf'] # 画像全般とPDFを許可
+)
+output_text_gr = gr.Textbox(
     label="Base64エンコード結果",
     lines=10,
-    interactive=True # 結果をコピーできるようにする
+    interactive=True # 結果をコピー可能にする
 )
 
 # Gradioインターフェースを作成
 iface = gr.Interface(
-    fn=encode_to_base64,       # 実行する関数
-    inputs=input_file,         # 入力コンポーネント
-    outputs=output_text,       # 出力コンポーネント
-    title="ファイル ⇨ Base64 エンコーダー",
+    fn=gradio_encode_handler,
+    inputs=input_file_gr,
+    outputs=output_text_gr,
+    title="ファイル ⇨ Base64 エンコーダー (Gradio UI)",
     description="画像 (JPG, PNG, GIF等) または PDF ファイルをアップロードすると、その Base64 エンコードされた文字列が表示されます。",
-    allow_flagging='never'     # Flagging機能が不要な場合は無効化
+    allow_flagging='never'
 )
 
-# アプリを起動
-iface.launch(share=False)
+# --- GradioアプリをFastAPIにマウント ---
+# Gradio UI をルートパス ("/") にマウントします
+app = gr.mount_gradio_app(app, iface, path="/")
+
+# --- ローカルでの実行用 (Hugging Face Spacesでは通常不要) ---
+# このブロックは `python app.py` で直接実行した場合のみ動作します。
+# Hugging Face Spaces環境では、Spacesがuvicornを起動してくれるため、この部分は実行されません。
+if __name__ == "__main__":
+    print("ローカルサーバーを起動します (http://127.0.0.1:8000)")
+    print("Gradio UI: http://127.0.0.1:8000/")
+    print("FastAPI エンドポイント (POST): http://127.0.0.1:8000/encode/")
+    print("API ドキュメント (Swagger UI): http://127.0.0.1:8000/docs")
+    print("API ドキュメント (ReDoc): http://127.0.0.1:8000/redoc")
+    # host="0.0.0.0" でローカルネットワーク内の他のデバイスからもアクセス可能に
+    uvicorn.run(app, host="0.0.0.0", port=8000)