Update app.py

app.py

@@ -106,14 +106,15 @@ class GitHubProxy:
         return any(white_item in url for white_item in Config.WHITE_LIST)
     
     @lru_cache(maxsize=1000)
-    def fetch_github_content(self, url: str, method: str = "GET") -> ProxyResponse:
+    def fetch_github_content(self, url: str, method: str = "GET", stream: bool = False) -> ProxyResponse:
         """获取GitHub内容（带缓存）"""
         try:
             response = self.session.request(
                 method=method,
                 url=url,
                 timeout=Config.TIMEOUT,
-                allow_redirects=False
+                allow_redirects=False,
+                stream=stream
             )
             
             headers = dict(response.headers)
@@ -130,9 +131,23 @@ class GitHubProxy:
                     redirect_url=redirect_url
                 )
             
+            # 如果是流式响应，直接返回response对象
+            if stream:
+                return ProxyResponse(
+                    status=response.status_code,
+                    content=response,  # 返回response对象以支持流式传输
+                    headers=headers
+                )
+            
+            # 检查是否是二进制内容
+            content_type = response.headers.get('content-type', '')
+            is_binary = not any(text_type in content_type.lower() for text_type in ['text', 'json', 'xml', 'html'])
+            
+            content = response.content if is_binary else response.text
+            
             return ProxyResponse(
                 status=response.status_code,
-                content=response.text,
+                content=content,
                 headers=headers
             )
             
@@ -222,10 +237,63 @@ def create_interface():
     """创建Gradio界面"""
     proxy = GitHubProxy()
     
-    with gr.Blocks(title="GitHub Proxy", theme=gr.themes.Soft()) as app:
+    def proxy_download(request: gr.Request):
+        """处理直接代理请求"""
+        # 从路径中提取GitHub URL
+        path = request.headers.get('x-path', '')
+        if not path:
+            return gr.Error("无效的请求路径")
+            
+        # 移除开头的斜杠
+        if path.startswith('/'):
+            path = path[1:]
+            
+        # 检查并处理URL
+        if not path.startswith(('http://', 'https://')):
+            path = 'https://' + path
+            
+        try:
+            # 获取代理响应
+            response = proxy.proxy_request(path, request)
+            
+            if 'error' in response:
+                return gr.Error(response['error'])
+                
+            if 'redirect_url' in response:
+                return gr.Redirect(response['redirect_url'])
+                
+            # 获取文件内容（使用流式传输）
+            proxy_response = proxy.fetch_github_content(path, stream=True)
+            if proxy_response.error:
+                return gr.Error(proxy_response.error)
+                
+            # 返回文件响应
+            headers = {
+                'Content-Type': proxy_response.headers.get('content-type', 'application/octet-stream'),
+                'Content-Disposition': proxy_response.headers.get('content-disposition', 'attachment'),
+            }
+            return gr.FileBinaryResponse(
+                proxy_response.content.raw,
+                headers=headers,
+                status=proxy_response.status
+            )
+                
+        except Exception as e:
+            logging.error(f"Proxy error: {str(e)}")
+            return gr.Error(f"代理请求失败: {str(e)}")
+    
+    # 添加直接代理路由
+    app = gr.App()
+    app.add_route("/.*", proxy_download, method="GET")
+    
+    with gr.Blocks(title="GitHub Proxy", theme=gr.themes.Soft()) as blocks:
         gr.Markdown("""
         # 🚀 GitHub Proxy
         
+        ### 使用方法
+        1. 直接访问: `https://your-domain.com/github-url`
+        2. 或者在下方输入GitHub URL进行测试
+        
         ### 功能特点
         - ✨ 支持多种GitHub URL格式
         - 🔄 自动处理重定向
@@ -282,11 +350,12 @@ def create_interface():
             examples=[
                 ["github.com/microsoft/vscode/blob/main/README.md"],
                 ["raw.githubusercontent.com/microsoft/vscode/main/README.md"],
-                ["gist.github.com/username/gist_id/raw/file.txt"]
+                ["github.com/ollama/ollama/releases/download/v0.5.1/ollama-windows-amd64.zip"]
             ],
             inputs=url_input
         )
     
+    app.blocks = blocks
     return app
 
 if __name__ == "__main__":