Upload folder using huggingface_hub

README.md

@@ -1,6 +1,6 @@
 ---
 title: sanatan_ai
-app_file: app.py
+app_file: main.py
 sdk: gradio
 sdk_version: 5.38.0
 python_version: 3.12

app.py

@@ -342,7 +342,7 @@ message_textbox = gr.Textbox(
 
 with gr.Blocks(
     theme=gr.themes.Citrus(),
-    title="Sanatan-Ai",
+    title="Sanatan-AI",
     css="""
     table {
     border-collapse: collapse;
@@ -407,7 +407,7 @@ with gr.Blocks(
         text-overflow: ellipsis;
     }
     """,
-) as app:
+) as gradio_app:
     show_sidebar = gr.State(True)
 
     # with gr.Column(scale=1, visible=show_sidebar.value) as sidebar_container:
@@ -468,4 +468,4 @@ with gr.Blocks(
         textbox=message_textbox,
     )
 
-app.launch()
+# app.launch()

main.py

@@ -1,22 +1,34 @@
-from langchain_openai import ChatOpenAI
-from dotenv import load_dotenv
-from push_notifications_helper import push
-from langchain.agents import Tool
+from fastapi.responses import RedirectResponse
+import uvicorn
+from fastapi import FastAPI
+from server import router as mobile_router
+from app import gradio_app  # your Blocks object
+import gradio as gr
+import logging
+from fastapi.middleware import Middleware
+from fastapi import Request
 
-from tools import tool_push, tool_search_web
+logging.basicConfig(level=logging.INFO)
 
-load_dotenv(override=True)
+app = FastAPI(title="Sanatan AI Unified Server")
 
+# Mount mobile endpoints
+app.include_router(mobile_router, prefix="/api")
 
-def main():
-    print("Hello from sanatan-ai!")
-    llm = ChatOpenAI()
-    llm_response = llm.invoke("Generate a simple question to ask the user about geography.")
-    # print(response)
-    response = tool_search_web.invoke(llm_response.content)
-    message = "Question: " + llm_response.content + "\nAnswer: " + response
-    tool_push.invoke(message)
+# Convert Gradio Blocks to ASGI app
+app = gr.mount_gradio_app(app, gradio_app,"/web")
 
+# Redirect root URL to /web/
+@app.get("/")
+async def redirect_to_web():
+    return RedirectResponse(url="/web/")
+
+@app.middleware("http")
+async def log_requests(request: Request, call_next):
+    logging.info(f"Request: {request.method} {request.url}")
+    response = await call_next(request)
+    logging.info(f"Response status: {response.status_code}")
+    return response
 
 if __name__ == "__main__":
-    main()
+    uvicorn.run("main:app", host="0.0.0.0", port=7860)

server.py

@@ -0,0 +1,29 @@
+# server.py
+import traceback
+import uuid
+from fastapi import APIRouter, Request
+from fastapi.responses import JSONResponse
+from gradio_client import Client
+from pydantic import BaseModel
+from app import chat
+
+router = APIRouter()
+
+
+class Message(BaseModel):
+    text: str
+
+
+@router.post("/chat")
+async def handle_chat(msg: Message):
+    try:
+        thread_id = uuid.uuid4()
+        reply_text = chat(
+            debug_mode=False, message=msg.text, history=None, thread_id=thread_id
+        )
+
+        return {"reply": reply_text}
+
+    except Exception as e:
+        traceback.print_exc()
+        return {"reply": f"Error: {e}"}