feat: HF deployment encahcement

gradio_chatbot.py

@@ -12,8 +12,14 @@ from loguru import logger
 env = environs.Env()
 env.read_env()
 
+# Hugging Face Spaces için özel yapılandırma
+IS_HF_SPACE = os.environ.get("SPACE_ID") is not None
+BASE_PATH = "" if not IS_HF_SPACE else "/spaces/lokumai/openai-openapi-template"
+
 # API Configuration
 BASE_URL = env.str("BASE_URL", "http://localhost:7860")
+if IS_HF_SPACE:
+    BASE_URL = f"https://huggingface.co{BASE_PATH}"
 API_KEY = env.str("API_KEY", "sk-test-xxx")
 CHAT_API_ENDPOINT = f"{BASE_URL}/v1/chat/completions"
 
@@ -26,79 +32,79 @@ USER_AVATAR = os.path.join(AVATAR_DIR, "user.png")
 BOT_AVATAR = os.path.join(AVATAR_DIR, "bot.png")
 
 # Custom CSS for fonts
-CUSTOM_CSS = """
-@font-face {
+CUSTOM_CSS = f"""
+@font-face {{
     font-family: 'UI Sans Serif';
-    src: url('/static/fonts/ui-sans-serif/ui-sans-serif-Regular.woff2') format('woff2');
+    src: url('{BASE_PATH}/static/fonts/ui-sans-serif/ui-sans-serif-Regular.woff2') format('woff2');
     font-weight: normal;
     font-style: normal;
-}
+}}
 
-@font-face {
+@font-face {{
     font-family: 'UI Sans Serif';
-    src: url('/static/fonts/ui-sans-serif/ui-sans-serif-Bold.woff2') format('woff2');
+    src: url('{BASE_PATH}/static/fonts/ui-sans-serif/ui-sans-serif-Bold.woff2') format('woff2');
     font-weight: bold;
     font-style: normal;
-}
+}}
 
-@font-face {
+@font-face {{
     font-family: 'System UI';
-    src: url('/static/fonts/system-ui/system-ui-Regular.woff2') format('woff2');
+    src: url('{BASE_PATH}/static/fonts/system-ui/system-ui-Regular.woff2') format('woff2');
     font-weight: normal;
     font-style: normal;
-}
+}}
 
-@font-face {
+@font-face {{
     font-family: 'System UI';
-    src: url('/static/fonts/system-ui/system-ui-Bold.woff2') format('woff2');
+    src: url('{BASE_PATH}/static/fonts/system-ui/system-ui-Bold.woff2') format('woff2');
     font-weight: bold;
     font-style: normal;
-}
+}}
 
-.gradio-container {
+.gradio-container {{
     font-family: 'UI Sans Serif', 'System UI', sans-serif;
-}
+}}
 
 /* Improve chat interface */
-.chat-message {
+.chat-message {{
     padding: 1rem;
     border-radius: 0.5rem;
     margin-bottom: 1rem;
     display: flex;
     align-items: flex-start;
-}
+}}
 
-.chat-message.user {
+.chat-message.user {{
     background-color: #f3f4f6;
-}
+}}
 
-.chat-message.bot {
+.chat-message.bot {{
     background-color: #eef2ff;
-}
+}}
 
 /* Improve button styles */
-button {
+button {{
     transition: all 0.2s ease-in-out;
-}
+}}
 
-button:hover {
+button:hover {{
     transform: translateY(-1px);
     box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
-}
+}}
 
 /* Improve input area */
-textarea {
+textarea {{
     border-radius: 0.5rem;
     padding: 0.75rem;
     border: 1px solid #e5e7eb;
     transition: border-color 0.2s ease-in-out;
-}
+}}
 
-textarea:focus {
+textarea:focus {{
     border-color: #4f46e5;
     outline: none;
     box-shadow: 0 0 0 2px rgba(79, 70, 229, 0.1);
-}
+}}
 """
 
 class MessageStatus(Enum):

main.py

@@ -10,6 +10,7 @@ from contextlib import asynccontextmanager
 from app.db.client import mongodb
 from gradio_chatbot import build_gradio_app
 import gradio as gr
+import os
 
 print(log_config.get_log_level())
 
@@ -18,6 +19,10 @@ env.read_env()
  
 STORAGE_TYPE = env.str("STORAGE_TYPE", "mongodb")
 
+# Hugging Face Spaces için özel yapılandırma
+IS_HF_SPACE = os.environ.get("SPACE_ID") is not None
+BASE_PATH = "" if not IS_HF_SPACE else "/spaces/lokumai/openai-openapi-template"
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """
@@ -35,7 +40,6 @@ async def lifespan(app: FastAPI):
     if STORAGE_TYPE == "mongodb":
         await mongodb.close()
 
-
 VERSION = "0.3.0"
 TITLE = "Talk to your data chat and visualize API"
 DESCRIPTION = """
@@ -76,6 +80,7 @@ app = FastAPI(
     lifespan= lifespan,
     openapi_tags= openapi_tags,
     debug= True,
+    root_path=BASE_PATH
 )
 
 # Configure OpenAPI security scheme
@@ -110,7 +115,7 @@ app = gr.mount_gradio_app(app, demo, path="/ui")
 @app.get("/")
 async def root():
     """Redirect root to Gradio UI"""
-    return RedirectResponse(url="/ui")
+    return RedirectResponse(url=f"{BASE_PATH}/ui")
 
 @app.get("/manifest.json")
 async def get_manifest():
@@ -119,18 +124,18 @@ async def get_manifest():
         "name": "Data Chatbot",
         "short_name": "Chatbot",
         "description": "A chatbot interface for data visualization and analysis",
-        "start_url": "/ui",
+        "start_url": f"{BASE_PATH}/ui",
         "display": "standalone",
         "background_color": "#ffffff",
         "theme_color": "#4f46e5",
         "icons": [
             {
-                "src": "/static/icons/icon-192x192.png",
+                "src": f"{BASE_PATH}/static/icons/icon-192x192.png",
                 "sizes": "192x192",
                 "type": "image/png"
             },
             {
-                "src": "/static/icons/icon-512x512.png",
+                "src": f"{BASE_PATH}/static/icons/icon-512x512.png",
                 "sizes": "512x512",
                 "type": "image/png"
             }