Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from langchain_mistralai.chat_models import ChatMistralAI
 from langchain.prompts import ChatPromptTemplate
 from langchain_deepseek import ChatDeepSeek
+from langchain_google_genai import ChatGoogleGenerativeAI
 import os
 from pathlib import Path
 import json
@@ -214,15 +215,24 @@ class OptimizedRAGLoader:
 #     streaming=True,
 # )
 
-deepseek_api_key = os.getenv("DEEPSEEK_KEY")
+# deepseek_api_key = os.getenv("DEEPSEEK_KEY")
+# llm = ChatDeepSeek(
+#     model="deepseek-chat",
+#     temperature=0,
+#     api_key=deepseek_api_key,
+#     streaming=True,
+# )
 
-llm = ChatDeepSeek(
-    model="deepseek-chat",
+
+gemini_api_key = os.getenv("GEMINI_KEY")
+llm = ChatGoogleGenerativeAI(
+    model="Gemini 2.0 Flash-Lite",
     temperature=0,
-    api_key=deepseek_api_key,
+    google_api_key=gemini_api_key,
     streaming=True,
 )
 
+
 rag_loader = OptimizedRAGLoader()
 retriever = rag_loader.get_retriever(k=5)  # Reduced k for faster retrieval