mrmtaeb commited on
Commit
d879ec7
·
verified ·
1 Parent(s): 8006e7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -1
app.py CHANGED
@@ -1 +1,23 @@
1
- # Placeholder for app.py (Unified Entrypoint)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from src.config.env import OPENAI_API_KEY, GOOGLE_API_KEY
2
+ from src.interface.ui import create_interface
3
+ from src.state import build_graph
4
+ from src.chains.qa_chain_openai import build_openai_rag_chain
5
+ from src.chains.qa_chain_gemini import build_gemini_rag_chain
6
+
7
+ # === Choose your backend model ===
8
+ MODEL = "openai" # Change to "gemini" to use Google Gemini
9
+ PDF_PATH = "resources/22_studenthandbook-22-23_f2.pdf"
10
+
11
+ # === Build chain based on chosen model ===
12
+ if MODEL == "openai":
13
+ rag_chain = build_openai_rag_chain(PDF_PATH)
14
+ elif MODEL == "gemini":
15
+ rag_chain = build_gemini_rag_chain(PDF_PATH)
16
+ else:
17
+ raise ValueError("Unsupported MODEL type. Choose 'openai' or 'gemini'.")
18
+
19
+ # === Build the state graph ===
20
+ app = build_graph(rag_chain)
21
+
22
+ # === Start the UI ===
23
+ create_interface(app.invoke)