Update app.py

app.py

@@ -2,8 +2,8 @@
 from fastapi import FastAPI, Request, Form
 from fastapi.templating import Jinja2Templates
 from fastapi.responses import HTMLResponse
-from fastapi.staticfiles import StaticFiles
 import wandb
+import os
 
 app = FastAPI()
 templates = Jinja2Templates(directory="./")
@@ -21,7 +21,11 @@ async def process_form(
     run_id: str = Form(...)
 ):
     try:
-        wandb.login(token)
+        # Set the token as an environment variable
+        os.environ["WANDB_API_KEY"] = token
+        # Login with the anonymous parameter set
+        wandb.login(key=token)
+        
         api = wandb.Api()
         run_path = f"{entity}/{project}/runs/{run_id}"
         run = api.run(run_path)
@@ -29,7 +33,11 @@ async def process_form(
         return templates.TemplateResponse(
             "index.html", 
             {
-                "request": request, 
+                "request": request,
+                "token": token,
+                "entity": entity,
+                "project": project,
+                "run_id": run_id,
                 "iframe_html": iframe_html
             }
         )
@@ -37,7 +45,11 @@ async def process_form(
         return templates.TemplateResponse(
             "index.html", 
             {
-                "request": request, 
+                "request": request,
+                "token": token,
+                "entity": entity,
+                "project": project,
+                "run_id": run_id,
                 "error": str(e)
             }
         )