Update app.py

app.py

@@ -1,7 +1,7 @@
 from src.config.env import OPENAI_API_KEY, GOOGLE_API_KEY, SERPER_API_KEY
 from src.interface.ui import create_interface
 from src.state import build_graph_with_callable
-from src.chains.qa_chain_openai import build_openai_rag_chain
+from src.chains.qa_chain_openai import build_openai_rag_chain_and_llm
 from src.chains.qa_chain_gemini import build_gemini_rag_chain
 from src.agents.prompts import CHARACTER_PROMPT
 from src.agents.tools import get_tools
@@ -15,7 +15,7 @@ PDF_PATH = "resources/22_studenthandbook-22-23_f2.pdf"
 
 # === Build chain based on chosen model ===
 if MODEL == "openai":
-    rag_chain = build_openai_rag_chain(PDF_PATH)
+    rag_chain, llm = build_openai_rag_chain_and_llm(PDF_PATH)
 elif MODEL == "gemini":
     rag_chain = build_gemini_rag_chain(PDF_PATH)
 else: