vikramvasudevan commited on
Commit
3279ad9
·
verified ·
1 Parent(s): 083e322

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. server.py +12 -0
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: