From abb4e450c6a448338f31a06b87b79830fb03635b Mon Sep 17 00:00:00 2001
From: Felipe Delestro Matos <fima@pop-os.localdomain>
Date: Thu, 30 Nov 2023 11:27:22 +0100
Subject: [PATCH] Hotfix to get the proper gradio port

---
 qim3d/utils/internal_tools.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/qim3d/utils/internal_tools.py b/qim3d/utils/internal_tools.py
index b168ac94..8a2362f0 100644
--- a/qim3d/utils/internal_tools.py
+++ b/qim3d/utils/internal_tools.py
@@ -272,15 +272,24 @@ def run_gradio_app(gradio_interface):
     host = "0.0.0.0"
     # Get port using the QIM API
     port_dict = get_port_dict()
-    gradio_header(gradio_interface.title, port_dict["port"])
+
+    if "gradio_port" in port_dict:
+        port = port_dict["gradio_port"]
+    elif "port" in port_dict:
+        port = port_dict["port"]
+    else:
+        raise Exception("Port not specified from QIM API")
+    
+    print(port_dict)
+    gradio_header(gradio_interface.title, port)
 
     # Create FastAPI with mounted gradio interface
     app = FastAPI()
-    path = f"/gui/{port_dict['username']}/{port_dict['port']}/"
+    path = f"/gui/{port_dict['username']}/{port}/"
     app = gr.mount_gradio_app(app, gradio_interface, path=path)
 
     # Full path
-    print(f"http://{host}:{port_dict['port']}{path}")
+    print(f"http://{host}:{port}{path}")
 
     # Run the FastAPI server usign uvicorn
-    run(app, host="0.0.0.0", port=int(port_dict["port"]))
+    run(app, host="0.0.0.0", port=int(port))
-- 
GitLab