update

.gitignore

@@ -0,0 +1,2 @@
+.env
+__pycache__/

app.py

@@ -1,7 +1,57 @@
-from fastapi import FastAPI
+# uvicorn app:app --host 0.0.0.0 --port 7860 --reload 
 
-app = FastAPI()
+from fastapi import FastAPI, Request, HTTPException
+import requests
+import os
+from dotenv import load_dotenv
+
+# 加载环境变量（存储Gemini API密钥）
+load_dotenv()
+app = FastAPI(title="Gemini Proxy API")
 
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
+
+# 1. 配置Gemini真实信息
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # 从环境变量获取密钥
+GEMINI_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/models"  # Gemini真实Base URL
+
+# 2. 通用转发接口（匹配Gemini的/predict等端点，路径动态匹配）
+@app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
+async def proxy_gemini(request: Request, path: str):
+    # 拼接Gemini真实请求URL（如客户端访问 /models/gemini-pro:generateContent → 转发到Gemini的对应路径）
+    gemini_url = f"{GEMINI_BASE_URL}/{path}?key={GEMINI_API_KEY}"
+    
+    # 提取客户端请求的headers和body，透传给Gemini
+    client_headers = dict(request.headers)
+    # 移除FastAPI自带的Host头，避免Gemini校验报错
+    client_headers.pop("host", None)
+    
+    try:
+        # 读取客户端请求体
+        client_body = await request.body()
+        
+        # 向Gemini发起请求（转发）
+        response = requests.request(
+            method=request.method,
+            url=gemini_url,
+            headers=client_headers,
+            data=client_body,
+            timeout=30  # 超时时间，可调整
+        )
+        
+        # 将Gemini的响应透传给客户端
+        return Response(
+            content=response.content,
+            status_code=response.status_code,
+            headers=dict(response.headers)
+        )
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"转发失败：{str(e)}")
+
+# 3. 健康检查接口（可选，用于验证服务是否正常）
+@app.get("/health")
+async def health_check():
+    return {"status": "ok", "proxy_target": "Google Gemini"}