Fix issue in server URL

airtai · Jun 11, 2024 · f4a208a · f4a208a
1 parent 1329ab1
commit f4a208a
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/weatherman/app.py b/weatherman/app.py
@@ -10,15 +10,14 @@
 logging.basicConfig(level=logging.INFO)
 
 host = environ.get("DOMAIN", "localhost")
-port = 9000
+port = 8000
 protocol = "http" if host == "localhost" else "https"
-
-app = FastAPI(
-    servers=[
-        {"url": f"{protocol}://{host}:{port}", "description": "Weather app server"}
-    ]
+base_url = (
+    f"{protocol}://{host}:{port}" if host == "localhost" else f"{protocol}://{host}"
 )
 
+app = FastAPI(servers=[{"url": base_url, "description": "Weather app server"}])
+
 
 class HourlyForecast(BaseModel):
     forecast_time: datetime.time