Staging

Pipfile

@@ -5,7 +5,7 @@ name = "pypi"
 
 [packages]
 fastapi = "~=0.116.1"
-uvicorn = "~=0.32.0"
+uvicorn = "~=0.35.0"
 litellm = "~=1.75.9"
 falkordb = "~=1.2.0"
 psycopg2-binary = "~=2.9.9"

api/app_factory.py

@@ -10,6 +10,8 @@
 from starlette.middleware.sessions import SessionMiddleware
 from starlette.middleware.base import BaseHTTPMiddleware
 
+from uvicorn.middleware.proxy_headers import ProxyHeadersMiddleware
+
 from api.auth.oauth_handlers import setup_oauth_handlers
 from api.auth.user_management import SECRET_KEY
 from api.routes.auth import auth_router, init_auth
@@ -49,7 +51,10 @@ def create_app():
             "Text2SQL with "
             "Graph-Powered Schema Understanding"
         ),
-    )
+    )    
+
+    app.add_middleware(ProxyHeadersMiddleware, trusted_hosts="*")
+
 
     # Add session middleware with explicit settings to ensure OAuth state persists
     app.add_middleware(

app/templates/base.j2

@@ -34,7 +34,7 @@
 
 <footer id="powered-by-falkordb">
   Powered by <a href="https://falkordb.com" target="_blank" rel="noopener">
-  <img src="{{ url_for('static', path='icons/logo.svg') }}" alt="FalkorDB Logo"/>
+  <img src="/static/icons/logo.svg" alt="FalkorDB Logo"/>
   </a>
 </footer>
 {% block scripts %}

app/templates/components/chat_header.j2

@@ -1,6 +1,6 @@
 {# Chat header with logo, title, and action buttons #}
 <div class="chat-header">
-    <img src="{{ url_for('static', path='icons/queryweaver.webp') }}" alt="Chat Logo" class="logo">
+    <img src="/static/icons/queryweaver.webp" alt="Chat Logo" class="logo">
     <h1>Natural Language to SQL Generator</h1>
     <div class="button-container">
         <select title="Select Database" id="graph-select">
@@ -20,11 +20,11 @@
                 </div>
                 <div class="dropdown-options" id="database-dropdown-options">
                     <div class="dropdown-option" data-value="postgresql">
-                        <img src="{{ url_for('static', path='icons/postgresql.svg') }}" alt="PostgreSQL" class="db-icon">
+                        <img src="/static/icons/postgresql.svg" alt="PostgreSQL" class="db-icon">
                         <span>PostgreSQL</span>
                     </div>
                     <div class="dropdown-option" data-value="mysql">
-                        <img src="{{ url_for('static', path='icons/mysql.svg') }}" alt="MySQL" class="db-icon">
+                        <img src="/static/icons/mysql.svg" alt="MySQL" class="db-icon">
                         <span>MySQL</span>
                     </div>
                 </div>

app/templates/components/chat_input.j2

@@ -3,10 +3,10 @@
     <div class="input-container" id="input-container">
         <input type="text" id="message-input" placeholder="Describe the SQL query you want..." />
         <button class="input-button" title="Submit" id="submit-button">
-            <img src="{{ url_for('static', path='icons/submit.svg') }}" alt="Submit">
+            <img src="/static/icons/submit.svg" alt="Submit">
         </button>
         <button class="input-button" title="Pause" id="pause-button">
-            <img src="{{ url_for('static', path='icons/pause.svg') }}" alt="Pause">
+            <img src="/static/icons/pause.svg" alt="Pause">
         </button>
         <button id="reset-button" class="action-button" title="Reset session">
             <svg width="800px" height="800px" viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg">

app/templates/components/login_modal.j2

@@ -4,10 +4,10 @@
         <h2>Welcome to QueryWeaver</h2>
         <p>Please login to continue</p>
         <a href="{{ url_for('google.login') }}" class="google-login-btn">
-            <img src="{{ url_for('static', path='icons/google.svg') }}"  alt="Google logo" class="google-login-logo">Sign in with Google
+            <img src="/static/icons/google.svg"  alt="Google logo" class="google-login-logo">Sign in with Google
         </a>
         <a href="{{ url_for('github.login') }}" class="github-login-btn">
-            <img src="{{ url_for('static', path='icons/github.svg') }}" alt="GitHub logo" class="github-login-logo">Sign in with GitHub
+            <img src="/static/icons/github.svg" alt="GitHub logo" class="github-login-logo">Sign in with GitHub
         </a>
     </div>
 </div>

app/templates/landing.j2

@@ -10,7 +10,7 @@
 <!-- Top header with logo -->
 <header class="site-header" role="banner">
   <div class="site-header-inner">
-    <img src="{{ url_for('static', path='icons/queryweaver.webp') }}" alt="QueryWeaver" />
+    <img src="/static/icons/queryweaver.webp" alt="QueryWeaver" />
   </div>
 </header>