Update app.py

app.py

@@ -1,87 +1,34 @@
-import httpx
-import asyncio
-import sys
-import types
+import requests
 import importlib.util
-from typing import Optional
+import sys
 
-class DynamicCodeExecutor:
-    def __init__(self, code_server_url: str):
-        """
-        动态代码执行器
+def load_and_run_from_txt(txt_url: str):
+    """同步方式下载txt并执行"""
+    try:
+        # 下载txt文件
+        response = requests.get(txt_url)
+        response.raise_for_status()
         
-        Args:
-            code_server_url: B服务器URL，如 "http://192.168.1.100:9000"
-        """
-        self.code_server_url = code_server_url.rstrip('/')
-        self.loaded_module = None
+        code_content = response.text
+        print(f"下载成功，代码长度: {len(code_content)}")
         
-    async def download_and_execute_code(self) -> bool:
-        """从B服务器下载代码并在内存中执行"""
-        try:
-            # 1. 下载源码
-            async with httpx.AsyncClient() as client:
-                response = await client.get(f"https://bg5-pycode.static.hf.space/api.txt")
-                result = response.text
-                
-            if not result["success"]:
-                print(f"下载代码失败: {result['error']}")
-                return False
-                
-            code_content = result["code"]
-            print("源码下载成功")
-            
-            # 2. 在内存中创建模块并执行
-            module_name = "dynamic_api"
-            spec = importlib.util.spec_from_loader(module_name, loader=None)
-            module = importlib.util.module_from_spec(spec)
-            
-            # 3. 执行代码
-            exec(code_content, module.__dict__)
-            
-            # 4. 将模块添加到sys.modules中，使其可被导入
-            sys.modules[module_name] = module
-            self.loaded_module = module
-            
-            print("代码执行成功，FastAPI应用已加载到内存")
-            return True
-            
-        except Exception as e:
-            print(f"执行代码失败: {e}")
-            return False
-    
-    def get_fastapi_app(self):
-        """获取动态加载的FastAPI应用实例"""
-        if self.loaded_module and hasattr(self.loaded_module, 'app'):
-            return self.loaded_module.app
-        return None
-    
-    async def start_server(self, host: str = "0.0.0.0", port: int = 8000):
-        """启动动态加载的FastAPI服务器"""
-        app = self.get_fastapi_app()
-        if not app:
-            print("FastAPI应用未加载")
-            return
+        # 创建并执行模块
+        spec = importlib.util.spec_from_loader("dynamic_api", loader=None)
+        module = importlib.util.module_from_spec(spec)
+        
+        exec(code_content, module.__dict__)
+        sys.modules["dynamic_api"] = module
+        
+        # 启动FastAPI
+        if hasattr(module, 'app'):
+            import uvicorn
+            uvicorn.run(module.app, host="0.0.0.0", port=8000)
+        else:
+            print("未找到FastAPI应用")
             
-        import uvicorn
-        config = uvicorn.Config(app, host=host, port=port)
-        server = uvicorn.Server(config)
-        await server.serve()
-
-# 使用示例
-async def main():
-    # 初始化动态代码执行器
-    executor = DynamicCodeExecutor("https://bg5-pycode.static.hf.space/api.txt")
-    
-    # 下载并执行代码
-    success = await executor.download_and_execute_code()
-    
-    if success:
-        # 启动服务器
-        print("启动FastAPI服务器...")
-        await executor.start_server(host="0.0.0.0", port=7860)
-    else:
-        print("代码加载失败")
+    except Exception as e:
+        print(f"执行失败: {e}")
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    # 直接执行
+    load_and_run_from_txt("https://bg5-pycode.static.hf.space/api.txt")