fileのstateを確認するように変更

app.py

@@ -143,30 +143,33 @@ def gemini(api_key, input_file, meeting_name, members, point, keyword, model, ta
         raise gr.Error("No api_key submitted!")
     if input_file is None:
         raise gr.Error("No input_file submitted! Please upload or record an file before submitting your request.")
-    
+
     # 音声 or 動画
     if task == "audio":
       system_prompt = audio_system_prompt(meeting_name, members, point, keyword)
-      
-      # 処理の実行
-      chat_session = model_settings(api_key, system_prompt, model)
-      logging.warning(f"input file: {input_file}")
-      upload_file = genai.upload_file(input_file)
-      logging.warning(f"upload file: {upload_file}")
-      time.sleep(15)
-      
     else:
-      
       system_prompt = video_system_prompt(meeting_name, members, point, keyword)
-      # 処理の実行
-      chat_session = model_settings(api_key, system_prompt, model)
-      logging.warning(f"input file: {input_file}")
-      upload_file = genai.upload_file(input_file)
-      logging.warning(f"upload file: {upload_file}")
-      time.sleep(45)
-
-    
-    
+
+
+    # 処理の実行
+    chat_session = model_settings(api_key, system_prompt, model)
+    logging.warning(f"input file: {input_file}")
+    upload_file = genai.upload_file(input_file)
+    logging.warning(f"upload file: {upload_file}")
+
+
+    # 処理の実行
+    # APIがファイルを受信したことを確認
+    while upload_file.state.name == "PROCESSING":
+        print('Waiting for processed.')
+        time.sleep(10)
+        upload_file = genai.get_file(upload_file.name)
+
+    if upload_file.state.name == "FAILED":
+        raise ValueError(upload_file.state.name)
+    print("upload processing complete:", upload_file.uri)
+    chat_session = model_settings(api_key, system_prompt, model)
+
     response = chat_session.send_message([upload_file])
 
     upload_file.delete()
@@ -183,7 +186,7 @@ audio = gr.Interface(
         gr.TextArea(label="参加者： 1名ずつ改行して入力", show_label=True, placeholder="不明"),
         gr.Textbox(label="抽出事項：　概要、何が議論されたのか、会議の結論、アクションアイテム以外に抽出したいこと", show_label=True),
         gr.Textbox(label="固有名詞：　企業名など", show_label=True),
-        gr.Radio(["gemini-1.5-flash-latest", "gemini-1.5-pro-latest"], label="使用モデル", show_label=True, value="gemini-1.5-flash-latest"),
+        gr.Radio(["gemini-1.5-flash-latest", "gemini-1.5-pro-latest"], label="使用モデル", show_label=True, value="gemini-1.5-pro-latest"),
         gr.Textbox(value="audio", visible=False),
     ],
     outputs=[gr.Chatbot(show_copy_button=True, render_markdown=True, height=1000, value=[[None, demo_markdown]])],