Vizznu19 commited on
Commit
937c1b4
·
verified ·
1 Parent(s): 735829b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -1,14 +1,13 @@
1
  import pandas as pd
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
4
- from fastapi.staticfiles import StaticFiles
5
  from pydantic import BaseModel
6
  from sentence_transformers import SentenceTransformer
7
  import faiss
8
  import numpy as np
9
  import re
10
 
11
- app = FastAPI()
12
 
13
  # Allow CORS for local frontend
14
  app.add_middleware(
@@ -19,9 +18,6 @@ app.add_middleware(
19
  allow_headers=["*"],
20
  )
21
 
22
- # Mount static files for Hugging Face Spaces
23
- app.mount("/", StaticFiles(directory="static", html=True), name="static")
24
-
25
  # Global variables for lazy initialization
26
  model = None
27
  chunk_index = None
@@ -65,6 +61,10 @@ class QueryRequest(BaseModel):
65
  query: str
66
  k: int = 1
67
 
 
 
 
 
68
  @app.post("/search")
69
  async def search_faq(req: QueryRequest):
70
  # Initialize model on first request
@@ -85,6 +85,7 @@ async def search_faq(req: QueryRequest):
85
  })
86
  return {"results": results}
87
 
88
- @app.get("/")
89
- def root():
90
- return {"message": "FAQ Assistant is running with chunking."}
 
 
1
  import pandas as pd
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
 
4
  from pydantic import BaseModel
5
  from sentence_transformers import SentenceTransformer
6
  import faiss
7
  import numpy as np
8
  import re
9
 
10
+ app = FastAPI(title="Bank FAQ Assistant", description="A semantic search FAQ system")
11
 
12
  # Allow CORS for local frontend
13
  app.add_middleware(
 
18
  allow_headers=["*"],
19
  )
20
 
 
 
 
21
  # Global variables for lazy initialization
22
  model = None
23
  chunk_index = None
 
61
  query: str
62
  k: int = 1
63
 
64
+ @app.get("/")
65
+ def root():
66
+ return {"message": "Bank FAQ Assistant is running. Use /search endpoint to query."}
67
+
68
  @app.post("/search")
69
  async def search_faq(req: QueryRequest):
70
  # Initialize model on first request
 
85
  })
86
  return {"results": results}
87
 
88
+ @app.get("/health")
89
+ def health_check():
90
+ return {"status": "healthy", "message": "FAQ Assistant is ready"}
91
+