Update app.py

update to handle google api key

app.py

@@ -1,3 +1,4 @@
+import os
 import serpapi
 import gradio as gr
 from langchain_google_genai import ChatGoogleGenerativeAI
@@ -5,7 +6,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from utils.utils import summarize_search_result, get_prompt
 
 
-def main(search_query, search_location, serp_api_key):
+def main(search_query, search_location, serp_api_key, google_api_key):
     search_response = serpapi.search(
         q=search_query,
         location=search_location,
@@ -16,6 +17,7 @@ def main(search_query, search_location, serp_api_key):
 
     prompt = get_prompt(search_query, search_summary)
 
+    os.environ["GOOGLE_API_KEY"] = google_api_key
     chat_model = ChatGoogleGenerativeAI(
         model="gemini-pro",
         convert_system_message_to_human=True,
@@ -34,7 +36,8 @@ if __name__ == "__main__":
             inputs=[
                 gr.Textbox(label="Search Query (検索クエリ)"),
                 gr.Textbox(label="Search Location (検索を行う国を英語で入力してください)"),
-                gr.Textbox(label="SERP API Key (https://serpapi.com/manage-api-key)")
+                gr.Textbox(label="SERP API Key (https://serpapi.com/manage-api-key)"),
+                gr.Textbox(label="Google API Key (https://console.cloud.google.com/apis/credentials)")
                 ],
             outputs=[gr.Textbox(label="Search Result (検索結果)")],
             title="Google Search enhanced by LLM"