Dheeraj-13 commited on
Commit
102f416
·
1 Parent(s): 0c3c572

Restore missing init_services function

Browse files
Files changed (1) hide show
  1. apps/web/app.py +19 -1
apps/web/app.py CHANGED
@@ -16,8 +16,26 @@ from services.rag.generate import get_generator, run_local_generation
16
  from services.rag.ingest import ingest
17
  from services.rag.index import build_index
18
  from services.observability.langfuse_client import observe
 
 
 
 
 
19
 
20
- # ...
 
 
 
 
 
 
 
 
 
 
 
 
 
21
 
22
  # GPU-wrapped generation function
23
  @spaces.GPU
 
16
  from services.rag.ingest import ingest
17
  from services.rag.index import build_index
18
  from services.observability.langfuse_client import observe
19
+ # Constants
20
+ DATA_DIR = "data"
21
+ PROCESSED_DIR = os.path.join(DATA_DIR, "processed")
22
+ INDEX_DIR = os.path.join(DATA_DIR, "index")
23
+ SAMPLE_DOCS_DIR = "sample_docs"
24
 
25
+ # Global Singletons
26
+ retriever = None
27
+ reranker = None
28
+ generator = None
29
+
30
+ def init_services():
31
+ global retriever, reranker, generator
32
+ try:
33
+ if os.path.exists(INDEX_DIR):
34
+ retriever = get_retriever(INDEX_DIR)
35
+ reranker = get_reranker()
36
+ generator = get_generator()
37
+ except Exception as e:
38
+ print(f"Service init warning: {e}")
39
 
40
  # GPU-wrapped generation function
41
  @spaces.GPU