Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py
+import re
 from fastapi import FastAPI, Request, Form
 from fastapi.templating import Jinja2Templates
 from fastapi.responses import HTMLResponse
@@ -30,6 +30,10 @@ async def process_form(
         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)
+        
         return templates.TemplateResponse(
             "index.html", 
             {