Upload 2 files

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from langchain_core.prompts import ChatPromptTemplate
-from langchain_openai import ChatOpenAI
+#from langchain_openai import ChatOpenAI
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langgraph.graph import StateGraph, END
 from typing import Dict, List
 
@@ -9,7 +10,14 @@ def agent_node(state):
     prompt = ChatPromptTemplate.from_messages(
         [("user", "{user_input}")]
     )
-    model = ChatOpenAI()
+    #model = ChatOpenAI()
+    model = ChatGoogleGenerativeAI(model="gemini-2.0-flash",
+        temperature=0,
+        max_tokens=None,
+        timeout=None,
+        max_retries=2,
+    # other params...
+    )
     response = model.invoke(prompt.format_messages(**state))
     return {"response": response.content}
 

requirements.txt

@@ -1,6 +1,7 @@
 aiofiles==23.2.1
 annotated-types==0.7.0
 anyio==4.9.0
+cachetools==5.5.2
 certifi==2025.1.31
 charset-normalizer==3.4.1
 click==8.1.8
@@ -8,11 +9,18 @@ distro==1.9.0
 fastapi==0.115.12
 ffmpy==0.5.0
 filelock==3.18.0
+filetype==1.2.0
 fsspec==2025.3.2
+google-ai-generativelanguage==0.6.17
+google-api-core==2.24.2
+google-auth==2.38.0
+googleapis-common-protos==1.69.2
 gradio==5.23.2
 gradio_client==1.8.0
 greenlet==3.1.1
 groovy==0.1.2
+grpcio==1.71.0
+grpcio-status==1.71.0
 h11==0.14.0
 httpcore==1.0.7
 httpx==0.28.1
@@ -24,6 +32,7 @@ jsonpatch==1.33
 jsonpointer==3.0.0
 langchain==0.3.22
 langchain-core==0.3.49
+langchain-google-genai==2.1.2
 langchain-openai==0.3.11
 langchain-text-splitters==0.3.7
 langgraph==0.3.21
@@ -41,6 +50,10 @@ ormsgpack==1.9.1
 packaging==24.2
 pandas==2.2.3
 pillow==11.1.0
+proto-plus==1.26.1
+protobuf==5.29.4
+pyasn1==0.6.1
+pyasn1_modules==0.4.2
 pydantic==2.11.1
 pydantic_core==2.33.0
 pydub==0.25.1
@@ -53,6 +66,7 @@ regex==2024.11.6
 requests==2.32.3
 requests-toolbelt==1.0.0
 rich==14.0.0
+rsa==4.9
 ruff==0.11.2
 safehttpx==0.1.6
 semantic-version==2.10.0