エラー処理追加、GUIからcsvアップロードを削除、サンプルの修正

app.py

@@ -19,14 +19,8 @@ GOOGLE_API_KEY=os.environ["GOOGLE_API_KEY"]
 
 def search_and_generate(question, prefix = "次の質問にできる限り答えてください。"):
     # ツールの準備
-    search = GoogleSearchAPIWrapper()
-    search_tool = Tool(
-        name="Search",
-        func=search.run,
-        description="useful for when you need to answer questions about current events"
-    ),
-
-    tools = load_tools(["google-search"], llm=ChatOpenAI())
+    search = GoogleSearchAPIWrapper(k=1)
+    tools = load_tools(["google-search","requests"], llm=ChatOpenAI())
 
     # プロンプトテンプレートの準備
     prefix = f"{prefix} 次のツールにアクセスできます:"
@@ -43,7 +37,7 @@ def search_and_generate(question, prefix = "次の質問にできる限り答え
     )
 
     # エージェントの準備
-    llm = ChatOpenAI(model_name="gpt-3.5-turbo")
+    llm = ChatOpenAI(model_name="gpt-4")
     llm_chain = LLMChain(llm=llm, prompt=prompt)
     agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools)
     agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)
@@ -52,7 +46,13 @@ def search_and_generate(question, prefix = "次の質問にできる限り答え
         result = agent_executor.run(question)
     except OutputParserException as e:
         result = '例外が発生しました: OutputParserException ' + str(e.args[0])
-    
+    except HttpError as e:
+        print("An HTTP error %d occurred:\n%s" % (e.resp.status, e.content))
+        # reasonがrateLimitExceededの場合
+        if "rateLimitExceeded" in e.content.decode():
+            result = "検索APIリクエストの上限に達しました。"
+        else:
+            result = "検索API実行時に通信エラーが発生しました。"
     return result
 
 def search_and_generate_csv(csv_file):
@@ -87,21 +87,12 @@ def search_and_generate_csv(csv_file):
     
     return result_csv
 
-def process_input(str_question, file_csv):
+def process_input(str_question, file_csv=None):
     # file_csvが空でない場合はCSVファイルとして扱う
-    try:
-        if file_csv:
-            return search_and_generate_csv(file_csv.name)
-        elif str_question:
-            return search_and_generate(str_question)
-    except HttpError as e:
-        print("An HTTP error %d occurred:\n%s" % (e.resp.status, e.content))
-        # reasonがrateLimitExceededの場合
-        if "rateLimitExceeded" in e.content.decode():
-            result = "検索APIリクエストの上限に達しました。"
-        else:
-            result = "検索API実行時に通信エラーが発生しました。"
-        return result
+    if file_csv:
+        return search_and_generate_csv(file_csv.name)
+    elif str_question:
+        return search_and_generate(str_question)
 
 def main():
     # パラメータに --csv がある場合、かつ、その次の引数がファイル名の場合
@@ -120,12 +111,12 @@ def main():
         print(result)
     else:
         gr.Interface(fn=process_input, 
-                 inputs=[gr.Text(), gr.File()], 
+                 inputs=[gr.Text()], 
                  outputs="text",
                  examples=[
-                            ["", "sample.csv"],
-                            ["日下部民藝館に伝わる伝承は？", None],
-                            ["醍醐寺に伝わる伝承は？", None],
+                            "ChatGPTの最新のアップデートは？",
+                            "日下部民藝館に伝わる伝承は？",
+                            "箱根に伝わる伝説は？",
                         ],
                  ).launch()
 

sample.csv

@@ -1,4 +1,7 @@
 question, answer
-葛飾区に伝わる伝承は？50文字以内で答えてください, 
-品川区に伝わる伝承は？50文字以内で答えてください, 
-箱根に伝わる伝承は？50文字以内で答えてください, 
+葛飾区に伝わる伝承は？ 
+品川区に伝わる伝承は？ 
+箱根に伝わる伝承は？ 
+日下部民藝館に伝わる伝承は？ 
+醍醐寺に伝わる伝承は？ 
+佐渡に伝わる伝承は？