Update app.py

app.py

@@ -1,34 +1,89 @@
 import gradio as gr
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse, Response
 import requests
 
-def proxy_request(path=""):
+app = FastAPI()
+
+@app.get("/api/{path:path}")
+@app.post("/api/{path:path}")
+async def proxy_api(path: str, request: Request):
     try:
-        # 构建目标 URL
+        # 构建 API 请求的目标 URL
         target_url = f"http://beibeioo.top/{path}"
         
-        # 发送请求
-        response = requests.get(target_url)
+        # 获取原始请求方法
+        method = request.method
+        
+        # 转发请求
+        response = requests.request(
+            method=method,
+            url=target_url,
+            headers={
+                'User-Agent': 'HuggingFace-Space-Proxy',
+                'Accept': '*/*'
+            }
+        )
         
-        # 返回响应内容
-        return response.json()
+        # 返回响应
+        return Response(
+            content=response.content,
+            status_code=response.status_code,
+            headers={
+                'Content-Type': response.headers.get('Content-Type', 'application/json'),
+                'Access-Control-Allow-Origin': '*',
+                'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
+                'Access-Control-Allow-Headers': '*'
+            }
+        )
     except Exception as e:
-        return {"error": str(e)}
+        return JSONResponse(
+            content={"error": str(e)},
+            status_code=500
+        )
 
-# 创建 Gradio 接口
-app = gr.Interface(
-    fn=proxy_request,
-    inputs=gr.Textbox(label="Path"),
-    outputs=gr.JSON(),
-    title="API Proxy"
-)
-
-# 添加自定义路由处理
-@app.queue()
-def proxy_all(request: gr.Request):
-    path = request.path
-    if path.startswith("/"):
-        path = path[1:]
-    return proxy_request(path)
+@app.get("/")
+@app.post("/")
+async def proxy_root(request: Request):
+    try:
+        # 直接访问 beibeioo.top
+        target_url = "http://beibeioo.top"
+        
+        # 获取原始请求方法
+        method = request.method
+        
+        # 转发请求
+        response = requests.request(
+            method=method,
+            url=target_url,
+            headers={
+                'User-Agent': 'HuggingFace-Space-Proxy',
+                'Accept': '*/*'
+            }
+        )
+        
+        # 返回响应
+        return Response(
+            content=response.content,
+            status_code=response.status_code,
+            headers={
+                'Content-Type': response.headers.get('Content-Type', 'application/json'),
+                'Access-Control-Allow-Origin': '*',
+                'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
+                'Access-Control-Allow-Headers': '*'
+            }
+        )
+    except Exception as e:
+        return JSONResponse(
+            content={"error": str(e)},
+            status_code=500
+        )
 
-if __name__ == "__main__":
-    app.launch()
+# 挂载 FastAPI 到 Gradio
+gr.mount_gradio_app(app, gr.Interface(
+    fn=lambda x: x,
+    inputs=gr.Textbox(visible=False),
+    outputs=gr.Textbox(visible=False),
+    examples=[],
+    title="API Proxy"
+), path="/ui")