Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Request, HTTPException, UploadFile, File, Form
+from fastapi import FastAPI, Request, HTTPException, UploadFile, File, Form, BackgroundTasks
 from pydantic import BaseModel
 import uuid
 from fastapi.responses import RedirectResponse
@@ -7,6 +7,7 @@ import os
 import requests
 from urllib.parse import urlencode
 import aiofiles
+import asyncio
 
 app = FastAPI()
 deployments = {}
@@ -14,7 +15,7 @@ deployments = {}
 # GitHub OAuth config (replace with your own client_id and client_secret)
 GITHUB_CLIENT_ID = os.getenv("GITHUB_CLIENT_ID", "your_github_client_id")
 GITHUB_CLIENT_SECRET = os.getenv("GITHUB_CLIENT_SECRET", "your_github_client_secret")
-GITHUB_OAUTH_REDIRECT = os.getenv("GITHUB_OAUTH_REDIRECT", "https://dragxd-host.hf.space/github/callback")
+GITHUB_OAUTH_REDIRECT = os.getenv("GITHUB_OAUTH_REDIRECT", "http://localhost:7860/github/callback")
 
 # In-memory user token storage (for demo; use DB in production)
 user_tokens = {}
@@ -25,6 +26,11 @@ user_selected_repo = {}
 # In-memory .env storage: {(user_id, repo_full_name): env_content}
 env_files = {}
 
+# In-memory log storage: {deploy_id: [log lines]}
+deploy_logs = {}
+# In-memory deploy status: {deploy_id: "deploying"|"success"|"failed"}
+deploy_status = {}
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -41,19 +47,32 @@ class RepoSelectRequest(BaseModel):
     user_id: str
     repo_full_name: str
 
+async def simulate_deployment(deploy_id, user_id, repo_full_name):
+    logs = [
+        f"Starting deployment for {repo_full_name}...",
+        "Cloning repository...",
+        "Installing dependencies...",
+        "Setting environment variables...",
+        "Running build...",
+        "Deploying application...",
+        "Finalizing...",
+        "Deployment successful!"
+    ]
+    deploy_logs[deploy_id] = []
+    deploy_status[deploy_id] = "deploying"
+    for log in logs:
+        deploy_logs[deploy_id].append(log)
+        await asyncio.sleep(2)  # Simulate time between log lines
+    deploy_status[deploy_id] = "success"
+
 @app.post("/deploy")
-async def deploy(req: DeployRequest):
-    # Simulate deployment
+async def deploy(req: DeployRequest, background_tasks: BackgroundTasks):
+    user_id = req.user_id
+    repo_full_name = user_selected_repo.get(user_id)
+    if not repo_full_name:
+        return {"error": "No repo selected for user."}
     deploy_id = str(uuid.uuid4())
-    deployments[deploy_id] = {
-        "github_url": req.github_url,
-        "user_id": req.user_id,
-        "status": "deploying",
-        "url": None
-    }
-    # Simulate deployment done after a few seconds (in real app, use background task)
-    deployments[deploy_id]["status"] = "success"
-    deployments[deploy_id]["url"] = f"https://demo-deployment/{deploy_id}"
+    background_tasks.add_task(simulate_deployment, deploy_id, user_id, repo_full_name)
     return {"deploy_id": deploy_id}
 
 @app.get("/status/{deploy_id}")
@@ -119,4 +138,15 @@ async def upload_env(user_id: str = Form(...), file: UploadFile = File(...)):
     if not repo_full_name:
         return {"error": "No repo selected for user."}
     env_files[(user_id, repo_full_name)] = content.decode()
-    return {"message": ".env uploaded", "repo_full_name": repo_full_name} 
+    return {"message": ".env uploaded", "repo_full_name": repo_full_name}
+
+@app.get("/logs/{deploy_id}")
+def get_logs(deploy_id: str, last_line: int = 0):
+    logs = deploy_logs.get(deploy_id, [])
+    status = deploy_status.get(deploy_id, "unknown")
+    # Return new log lines since last_line
+    return {
+        "logs": logs[last_line:],
+        "next_line": len(logs),
+        "status": status
+    }