Update app.py

app.py

@@ -1,132 +1,71 @@
 import gradio as gr
 import base64
-import os
-import io
-from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
-import uvicorn # ローカル実行用にインポート
+import uvicorn # FastAPIアプリを起動するために必要
 
-# --- Base64 エンコード処理 (共通ロジック) ---
-def encode_bytes_to_base64(file_bytes: bytes) -> str:
-    """
-    バイト列を受け取り、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}")
+# --- FastAPI アプリケーションの作成 ---
+app = FastAPI()
 
-# --- Gradio用 関数 ---
-def gradio_encode_handler(file_obj):
+# --- (既存の) Base64 エンコード関数 ---
+def encode_to_base64(file_obj):
     """
-    GradioのFileコンポーネントから受け取ったファイルを処理する関数。
+    gr.FileまたはUploadFileから受け取ったファイルオブジェクトを処理し、
+    Base64エンコードされた文字列を返す関数。
     """
     if file_obj is None:
         return "ファイルをアップロードしてください。"
 
     try:
-        file_path = file_obj.name
-        print(f"Gradio処理中のファイル: {file_path}") # デバッグ用ログ
+        if isinstance(file_obj, UploadFile): # FastAPIのUploadFileの場合
+            file_bytes = await file_obj.read() # await を使用
+        else: # GradioのFileの場合 (file_obj.name はファイルパス)
+            file_path = file_obj.name
+            with open(file_path, 'rb') as f:
+                file_bytes = f.read()
 
-        with open(file_path, 'rb') as f:
-            file_bytes = f.read()
-
-        # 共通のエンコード関数を呼び出す
-        base64_string = encode_bytes_to_base64(file_bytes)
+        base64_bytes = base64.b64encode(file_bytes)
+        base64_string = base64_bytes.decode('utf-8')
         return base64_string
 
     except Exception as e:
-        print(f"Gradioファイル処理エラー: {e}") # デバッグ用ログ
-        # Gradioではエラーメッセージを文字列として返す
+        print(f"エラー発生: {e}")
         return f"ファイルの処理中にエラーが発生しました: {e}"
 
-# --- FastAPI アプリケーションの初期化 ---
-app = FastAPI(
-    title="File to Base64 API & Gradio UI",
-    description="ファイルを受け取りBase64にエンコードするAPIとGradio UIを提供します。"
-)
-
-# --- 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 エンドポイント ---
+@app.post("/api/encode_base64")
+async def api_encode_base64(file: UploadFile = File(...)):
     """
-    FastAPIのエンドポイント。
-    POSTリクエストでファイルを受け取り、Base64エンコードしてJSONで返す。
+    APIエンドポイント: ファイルを受け取り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="空のファイルは処理できません。")
-
-    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()
+    base64_string = await encode_to_base64(file) # await を使用
+    if "エラー" in base64_string: # エラーメッセージの場合
+        return JSONResponse(content={"error": base64_string}, status_code=400) # エラーレスポンス
+    return {"base64_string": base64_string} # 成功レスポンス
 
-# --- Gradio インターフェースの定義 ---
-input_file_gr = gr.File(
+# --- Gradio インターフェース (変更なし) ---
+input_file = gr.File(
     label="画像またはPDFファイルを入力",
-    file_types=['image', '.pdf'] # 画像全般とPDFを許可
+    file_types=['image', '.pdf']
 )
-output_text_gr = gr.Textbox(
-    label="Base64エンコード結果",
+output_text = gr.Textbox(
+    label="Base64エンコード結果 (Gradio)",
     lines=10,
-    interactive=True # 結果をコピー可能にする
+    interactive=True
 )
 
-# Gradioインターフェースを作成
 iface = gr.Interface(
-    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_mode="never"  # Flagging機能を無効にする（推奨される書き方）
+    fn=encode_to_base64,
+    inputs=input_file,
+    outputs=output_text,
+    title="ファイル ⇨ Base64 エンコーダー (Gradio & API)",
+    description="画像またはPDFファイルをアップロードしてBase64エンコード。\nAPIエンドポイント: `/api/encode_base64` (POST)",
+    allow_flagging='never'
 )
 
-# --- GradioアプリをFastAPIにマウント ---
-# Gradio UI をルートパス ("/") にマウントします
-app = gr.mount_gradio_app(app, iface, path="/")
+# --- アプリケーションの起動 ---
+# GradioアプリとFastAPIアプリを両方起動 (uvicorn経由でFastAPIを起動)
+app = gr.mount_gradio_app(app, iface, path="/gradio") # GradioアプリをFastAPIにマウント
 
-# --- ローカルでの実行用 (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)
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000) # FastAPIアプリをuvicornで起動