Update app.py

Update to get google api key and cse id from user input

app.py

@@ -1,25 +1,22 @@
+import os
 from dotenv import load_dotenv
 import gradio as gr
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents import load_tools, initialize_agent
 
 
-# Load environment variables
-_ = load_dotenv()
-
-
-llm = ChatGoogleGenerativeAI(
-    model="gemini-pro",
-    temperature=0.0, # temperature=0.7 (default)
-    # top_p=0.5,
-    )
-tools = load_tools(["google-search"], llm=llm)
-
-
-agent = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True)
+def main(query, google_api_key, google_cse_id):
+    # Set environment variables
+    os.environ["GOOGLE_API_KEY"] = google_api_key
+    os.environ["GOOGLE_CSE_ID"] = google_cse_id
 
+    llm = ChatGoogleGenerativeAI(
+        model="gemini-pro",
+        temperature=0.0
+        )
+    tools = load_tools(["google-search"], llm=llm)
+    agent = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True)
 
-def main(query):
     return agent.run(query)
 
 
@@ -27,8 +24,11 @@ if __name__ == "__main__":
     try:
         app = gr.Interface(
             fn=main,
-            # inputs=["text", "text"],
-            inputs=[gr.Textbox(label="Search Query (クエリを入力してください)")],
+            inputs=[
+                gr.Textbox(label="Search Query (検索クエリ)"),
+                gr.Textbox(label="Google API Key (https://console.cloud.google.com/apis/credentials)"),
+                gr.Textbox(label="Google Programmable Search Engine (https://programmablesearchengine.google.com)")
+                ],
             outputs=[gr.Textbox(label="Search Result (検索結果)")],
             title="Google Search enhanced by LLM"
         )