Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Upload folder using huggingface_hub
Browse files
server.py
CHANGED
|
@@ -5,6 +5,7 @@ from fastapi import APIRouter, Request
|
|
| 5 |
from fastapi.responses import JSONResponse
|
| 6 |
from pydantic import BaseModel
|
| 7 |
from app import chat
|
|
|
|
| 8 |
|
| 9 |
router = APIRouter()
|
| 10 |
|
|
@@ -16,6 +17,17 @@ class Message(BaseModel):
|
|
| 16 |
text: str
|
| 17 |
session_id: str | None = None # Optional session ID from client
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
@router.post("/chat")
|
| 20 |
async def handle_chat(msg: Message, request: Request):
|
| 21 |
try:
|
|
|
|
| 5 |
from fastapi.responses import JSONResponse
|
| 6 |
from pydantic import BaseModel
|
| 7 |
from app import chat
|
| 8 |
+
from config import SanatanConfig
|
| 9 |
|
| 10 |
router = APIRouter()
|
| 11 |
|
|
|
|
| 17 |
text: str
|
| 18 |
session_id: str | None = None # Optional session ID from client
|
| 19 |
|
| 20 |
+
@router.post("/greet")
|
| 21 |
+
async def handle_greet(msg: Message):
|
| 22 |
+
markdown = "Namaskaram 🙏 I am Bhashyam.AI and I can help you explore the following scriptures:\n---\n"
|
| 23 |
+
for scripture in SanatanConfig().scriptures:
|
| 24 |
+
markdown += f"- {scripture['title']}\n"
|
| 25 |
+
|
| 26 |
+
session_id = msg.session_id
|
| 27 |
+
if not session_id:
|
| 28 |
+
session_id = str(uuid.uuid4())
|
| 29 |
+
return {"reply": markdown, "session_id": session_id}
|
| 30 |
+
|
| 31 |
@router.post("/chat")
|
| 32 |
async def handle_chat(msg: Message, request: Request):
|
| 33 |
try:
|