Update app.py

app.py

@@ -14,6 +14,8 @@ from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.utilities import WikipediaAPIWrapper
 from langchain_community.document_loaders import ArxivLoader
+import google.generativeai as genai
+
 
 # (Keep Constants as is)
 # --- Constants ---