add progress bar

app.py

@@ -186,12 +186,15 @@ def fetch_clinical_trials_basic(cancer_name):
         return []
 
 # 軽量版データ生成関数
-def generate_sample_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable):
+def generate_sample_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable, progress=None):
     """サンプルデータを生成（辞書のリスト形式）"""
     try:
         if not all([age, sex, tumor_type]):
             return []
         
+        if progress:
+            progress(0.3, desc="📋 サンプルデータを生成中...")
+        
         sample_data = [
             {
                 "NCTID": "NCT12345678",
@@ -228,6 +231,9 @@ def generate_sample_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiabl
             }
         ]
         
+        if progress:
+            progress(1.0, desc="✅ サンプルデータ生成完了")
+        
         return sample_data
         
     except Exception as e:
@@ -235,15 +241,21 @@ def generate_sample_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiabl
         return []
 
 # 基本版データ生成関数（ClinicalTrials.gov API使用、AI評価なし）
-def generate_basic_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable):
+def generate_basic_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable, progress=None):
     """基本版のデータ生成（API使用、AI評価なし）"""
     try:
         if not all([age, sex, tumor_type]):
             return []
         
+        if progress:
+            progress(0.2, desc="🔍 ClinicalTrials.govからデータを検索中...")
+        
         # 実際のAPI呼び出し
         data_list = fetch_clinical_trials_basic(tumor_type)
         
+        if progress:
+            progress(0.7, desc="📊 データを処理中...")
+        
         if not data_list:
             print("臨床試験データが見つかりませんでした")
             return []
@@ -253,6 +265,9 @@ def generate_basic_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable
             item['AgentJudgment'] = f'基本版：{age}歳{sex}の{tumor_type}患者への詳細評価にはAI機能が必要です'
             item['AgentGrade'] = 'unclear'
         
+        if progress:
+            progress(1.0, desc=f"✅ 完了: {len(data_list)}件の臨床試験を取得")
+        
         print(f"基本版評価完了。結果: {len(data_list)} 件")
         return data_list
         
@@ -262,15 +277,20 @@ def generate_basic_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable
         return []
 
 # 完全版データ生成関数（AI評価付き）
-def generate_full_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable):
+def generate_full_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable, progress=None):
     """完全版のデータ生成（実際のAPI使用 + AI評価）"""
     try:
         if not all([age, sex, tumor_type]):
             return []
         
+        if progress:
+            progress(0.1, desc="🔄 AI評価システムを初期化中...")
+        
         # 日本語の腫瘍タイプを英語に翻訳
         try:
             if translator is not None:
+                if progress:
+                    progress(0.15, desc="🌐 腫瘍タイプを英語に翻訳中...")
                 TumorName = translator.translate(tumor_type)
                 print(f"腫瘍タイプ翻訳: {tumor_type} → {TumorName}")
             else:
@@ -282,6 +302,8 @@ def generate_full_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable)
 
         # 質問文を生成
         try:
+            if progress:
+                progress(0.2, desc="❓ 患者情報から評価用質問を生成中...")
             ex_question = generate_ex_question_English(age, sex, TumorName, GeneMutation, Meseable, Biopsiable)
             print(f"生成された質問: {ex_question}")
         except Exception as e:
@@ -290,6 +312,8 @@ def generate_full_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable)
         
         # 臨床試験データの取得
         try:
+            if progress:
+                progress(0.3, desc="🔍 ClinicalTrials.govから臨床試験データを検索中...")
             print(f"臨床試験データを検索中: {TumorName}")
             df = fetch_clinical_trials(TumorName)
             if df.empty:
@@ -308,8 +332,16 @@ def generate_full_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable)
         evaluation_limit = min(len(data_list), 10)
         print(f"AI評価実行: {evaluation_limit} 件を処理します")
         
+        if progress:
+            progress(0.4, desc=f"🤖 AI評価開始: {evaluation_limit}件の臨床試験を分析中...")
+        
         for i, item in enumerate(data_list[:evaluation_limit]):
             try:
+                # プログレスバーの更新（0.4から0.9まで）
+                progress_value = 0.4 + (0.5 * (i + 1) / evaluation_limit)
+                if progress:
+                    progress(progress_value, desc=f"🧠 AI評価中: {i+1}/{evaluation_limit} - {item['NCTID']}")
+                
                 print(f"評価中 ({i+1}/{evaluation_limit}): {item['NCTID']}")
                 target_criteria = item['Eligibility Criteria']
                 
@@ -327,10 +359,16 @@ def generate_full_data(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable)
                 item['AgentGrade'] = "unclear"
         
         # 評価されなかった残りのアイテムにはプレースホルダーを設定
+        if progress:
+            progress(0.95, desc="📊 結果を整理中...")
+        
         for item in data_list[evaluation_limit:]:
             item['AgentJudgment'] = f"完全版：{age}歳{sex}の{tumor_type}患者（評価制限により未処理）"
             item['AgentGrade'] = "unclear"
         
+        if progress:
+            progress(1.0, desc=f"✅ 完了: {len(data_list)}件中{evaluation_limit}件をAI評価")
+        
         print(f"完全版評価完了。結果: {len(data_list)} 件（うち{evaluation_limit}件をAI評価）")
         return data_list
         
@@ -585,30 +623,32 @@ with gr.Blocks(title="臨床試験適格性評価", theme=gr.themes.Soft()) as d
     progress_text = gr.Textbox(label="Processing Status", value="Ready", interactive=False)
 
     # イベントハンドリング
-    def update_data_and_display(age, sex, tumor_type, gene_mutation, measurable, biopsiable):
+    def update_data_and_display(age, sex, tumor_type, gene_mutation, measurable, biopsiable, progress=gr.Progress()):
         """データ生成と表示更新"""
         try:
             groq_available = bool(os.getenv("GROQ_API_KEY"))
             
+            progress(0.05, desc="🚀 処理を開始しています...")
+            
             if FULL_VERSION and groq_available:
-                progress_msg = "🤖 AI適格性評価付きで実際の臨床試験データを検索中..."
-                data = generate_full_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable)
+                progress(0.1, desc="🤖 AI適格性評価モードで実行中...")
+                data = generate_full_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable, progress)
             elif FULL_VERSION:
-                progress_msg = "🔍 実際の臨床試験データを検索中（AI評価にはGROQ_API_KEY必要）..."
-                data = generate_basic_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable)
+                progress(0.1, desc="🔍 基本モードで実行中...")
+                data = generate_basic_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable, progress)
             elif LANGCHAIN_AVAILABLE:
-                progress_msg = "📡 ClinicalTrials.govから基本データを検索中..."
-                data = generate_basic_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable)
+                progress(0.1, desc="📡 基本検索モードで実行中...")
+                data = generate_basic_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable, progress)
             else:
-                progress_msg = "📋 サンプルデータを生成中..."
-                data = generate_sample_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable)
+                progress(0.1, desc="📋 サンプルモードで実行中...")
+                data = generate_sample_data(age, sex, tumor_type, gene_mutation, measurable, biopsiable, progress)
             
             if data:
                 html_table = create_html_table(data)
                 final_progress = f"✅ 完了: {len(data)} 件の臨床試験が見つかりました"
-                if FULL_VERSION and env_ok:
-                    ai_count = len([item for item in data if 'エラー' not in item.get('AgentJudgment', '')])
-                    final_progress += f"（うち最大10件をAI評価済み）"
+                if FULL_VERSION and groq_available:
+                    ai_evaluated_count = len([item for item in data if 'エラー' not in item.get('AgentJudgment', '') and '未処理' not in item.get('AgentJudgment', '')])
+                    final_progress += f"（うち{min(len(data), 10)}件をAI評価済み）"
             else:
                 html_table = "<div style='text-align: center; padding: 20px; color: #666;'>⚠️ 該当する臨床試験が見つかりませんでした</div>"
                 final_progress = "⚠️ 該当する臨床試験が見つかりませんでした"