Update app.py

app.py

@@ -1,18 +1,42 @@
-import os
-from app.main import app
-import uvicorn
 from fastapi import FastAPI
-app = FastAPI()
+from fastapi.middleware.cors import CORSMiddleware
+import uvicorn
+
+# Create FastAPI app
+app = FastAPI(
+    title="AI-Powered Test Automation Framework",
+    description="An intelligent test automation framework powered by AI",
+    version="1.0.0"
+)
 
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Health check endpoint
 @app.get("/")
 async def root():
     return {
-        "name": "Test",
-        "version": "1.0.0",
-        "status": "running"
+        "message": "Welcome to AI-Powered Test Automation Framework",
+        "status": "healthy",
+        "docs_url": "/docs"
     }
 
+# Import and include other routers
+from api.routes import document_routes, ai_routes, test_routes, automation_routes
+
+app.include_router(document_routes.router, prefix="/api/documents", tags=["Documents"])
+app.include_router(ai_routes.router, prefix="/api/ai", tags=["AI"])
+app.include_router(test_routes.router, prefix="/api/tests", tags=["Tests"])
+app.include_router(automation_routes.router, prefix="/api/automation", tags=["Automation"])
+
+# Run the application
 if __name__ == "__main__":
-    # Get port from environment variable (for Hugging Face Spaces)
+    import os
     port = int(os.getenv("PORT", 7860))
-    uvicorn.run(app, host="0.0.0.0", port=port)   
+    uvicorn.run(app, host="0.0.0.0", port=port)