Update app.py

app.py

@@ -1,9 +1,10 @@
-import re
+# app.py
 from fastapi import FastAPI, Request, Form
 from fastapi.templating import Jinja2Templates
 from fastapi.responses import HTMLResponse
 import wandb
 import os
+from typing import Optional
 
 app = FastAPI()
 templates = Jinja2Templates(directory="./")
@@ -15,24 +16,22 @@ async def index(request: Request):
 @app.post("/", response_class=HTMLResponse)
 async def process_form(
     request: Request,
-    token: str = Form(...),
+    token: Optional[str] = Form(None),  # Make token optional
     entity: str = Form(...),
     project: str = Form(...),
     run_id: str = Form(...)
 ):
     try:
-        # Set the token as an environment variable
-        os.environ["WANDB_API_KEY"] = token
-        # Login with the anonymous parameter set
-        wandb.login(key=token)
-        
+        # Only set token and login if one is provided
+        if token and token.strip():
+            os.environ["WANDB_API_KEY"] = token
+        else:
+                    
         api = wandb.Api()
         run_path = f"{entity}/{project}/runs/{run_id}"
         run = api.run(run_path)
         iframe_html = run.to_html()
-
-        # Modify the iframe height to be 100%
-        iframe_html = re.sub(r'height:\d+px', 'height:100%', iframe_html)
+        iframe_html = iframe_html.replace('height:420px', 'height:100%')
         
         return templates.TemplateResponse(
             "index.html",