Update app.py

app.py

@@ -2,55 +2,60 @@ import gradio as gr
 import re
 import os
 import base64
+import time  # 進捗バー用
 from datetime import datetime, timedelta, timezone
 
 # 設定
-SUPPORT_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/main/README.md"#Githubのメインページ
-HELP_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/Q%26A/README.md"#GithubのFAQ
-COPYRIGHT_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/copyright/README.md"#Githubのコピーライトページ
-CONTACT_URL = "https://forms.gle/mAUj1CdhufHFbiWs7"#Googleフォーム
-TS_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/troubleshooting/README.md"#Githubのトラブルシューティングページ(TS=トラブルシューティング)
+SUPPORT_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/main/README.md"
+HELP_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/Q%26A/README.md"
+COPYRIGHT_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/copyright/README.md"
+CONTACT_URL = "https://forms.gle/mAUj1CdhufHFbiWs7"
+TS_URL = "https://github.com/X1288664/LINE-Log-Manager-for-Hugging-face/blob/troubleshooting/README.md"
 JST = timezone(timedelta(hours=9))
 
-def load_log_file(file):
-    """アップロードされたログファイルを開き、内容を取得"""
+def load_log_file(file, progress=gr.Progress()):
+    """アップロードされたログファイルを開き、内容を取得 + プログレスバー"""
     if not file:
         return None, "エラー(FILE-001): ファイルが選択されていません"
 
-         # ファイルの拡張子をチェック (.txt 以外はエラー)
     if not file.name.endswith(".txt"):
         return None, "エラー(FILE-005): サポートされていないファイル形式です。TXTファイルを選択してください。"
-    
+
     try:
         with open(file.name, "r", encoding="utf-8") as f:
-            content = f.readlines()
-        return content, None
+            lines = f.readlines()
+
+        # プログレスバー（ログファイル読み込み）
+        for i in progress.tqdm(range(1, 101), desc="ログファイルを読み込み中..."):
+            time.sleep(0.01)  # 処理シミュレーション
+
+        return lines, None
     except Exception as e:
         return None, "エラー(FILE-004): ファイルのエンコード形式が不正です(UTF-8のみ対応)"
 
-def search_logs(file, keyword, year, month, day):
-    """ログを検索し、結果を返す"""
+def search_logs(file, keyword, year, month, day, progress=gr.Progress()):
+    """ログ検索処理 + プログレスバー"""
     log_lines, error = load_log_file(file)
     if error:
-        return "", "エラー(FILE-002): ファイルの読み込みに失敗しました", None
+        return "", error, None
 
     if not log_lines:
         return "", "エラー(FILE-003): ファイルが空です。", None
 
-    # 検索条件のチェック
     is_word_search = bool(keyword.strip())
     is_date_search = year != "----" and month != "----" and day != "----"
 
     if is_word_search and is_date_search:
         return "", "エラー(SEARCH-001): ワード検索と日付検索は同時に行えません。", None
 
-    # 検索条件の設定
     search_condition = "検索条件: なし"
     results = []
 
     if is_word_search:
         search_condition = f"検索条件: ワード[{keyword}]"
-        results = [f"- {line.strip()}" for line in log_lines if keyword in line]
+        for line in progress.tqdm(log_lines, desc="検索中..."):
+            if keyword in line:
+                results.append(f"- {line.strip()}")
 
     elif is_date_search:
         try:
@@ -64,39 +69,16 @@ def search_logs(file, keyword, year, month, day):
             return "", "エラー(SEARCH-002): 無効な日付形式です。", None
 
     formatted_results = "\n".join(results) if results else "SEARCH-003: 一致する結果が見つかりませんでした。"
-
     return formatted_results, f"{len(results)} 件の結果が見つかりました。", search_condition
 
-def extract_date_range_logs(lines, start_date, end_date):
-    """指定された日付範囲のログを抽出する"""
-    results = []
-    current_date = None
-
-    for line in lines:
-        date_match = re.match(r"(\d{4}/\d{1,2}/\d{1,2})", line)
-        if date_match:
-            try:
-                current_date = datetime.strptime(date_match.group(1), "%Y/%m/%d")
-            except ValueError:
-                continue
-
-        if current_date and start_date <= current_date < end_date:
-            results.append(line.strip())
-
-    return results
-
-def generate_download_link(results, search_condition):
-    """検索結果をTXTファイルとしてbase64エンコードし、ダウンロードリンクを生成"""
+def generate_download_link(results, search_condition, progress=gr.Progress()):
+    """エクスポート処理 + プログレスバ��"""
     if not results:
         return "エラー(EXPORT-001): エクスポートするデータがありません。"
 
-    #エクスポートファイル名に挿入する用
     execution_time = datetime.now(JST).strftime("%Y%m%d_%H%M%S")
-    
-    #エクスポートファイル内の検索実行日に表記する用
     JPN_date = datetime.now(JST).strftime("%Y/%m/%d %H:%M:%S")
-    
-    # 検索条件をファイル名に適用
+
     if "ワード" in search_condition:
         condition_text = search_condition.replace("検索条件: ワード[", "").replace("]", "").replace(" ", "_")
     elif "日付" in search_condition:
@@ -116,13 +98,13 @@ def generate_download_link(results, search_condition):
     )
 
     encoded_file = base64.b64encode(file_content.encode()).decode()
-    href = f'<a href="data:text/plain;base64,{encoded_file}" download="{file_name}">結果をダウンロード</a>'
-    return href
 
-    if not encoded_file:
-        return "エラー(EXPORT-002): エクスポートファイルの作成に失敗しました"
+    # エクスポートの進捗を表示
+    for i in progress.tqdm(range(1, 101), desc="エクスポート処理中..."):
+        time.sleep(0.01)
 
-    
+    href = f'<a href="data:text/plain;base64,{encoded_file}" download="{file_name}">結果をダウンロード</a>'
+    return href
 
 # Gradio UI
 with gr.Blocks() as demo:
@@ -130,10 +112,8 @@ with gr.Blocks() as demo:
 
     file_input = gr.File(label="ログファイルをアップロード")
 
-    # ユーザー検索用テキストボックス
     search_word = gr.Textbox(label="ユーザー検索ワード")
 
-    # 日付検索用ドロップダウン
     year = gr.Dropdown(choices=["----"] + [str(y) for y in range(2020, 2041)], value="----", label="年")
     month = gr.Dropdown(choices=["----"] + [str(m) for m in range(1, 13)], value="----", label="月")
     day = gr.Dropdown(choices=["----"] + [str(d) for d in range(1, 32)], value="----", label="日")
@@ -144,9 +124,8 @@ with gr.Blocks() as demo:
 
     search_condition_state = gr.State()
 
-    # エクスポート機能
     export_button = gr.Button("TXTをエクスポートする")
-    export_output = gr.HTML()  # base64エンコードしたリンクを表示
+    export_output = gr.HTML()
     export_status = gr.Textbox(label="エクスポートメッセージ", interactive=False)
 
     search_button.click(
@@ -167,5 +146,4 @@ with gr.Blocks() as demo:
     gr.Markdown(f"[🛠 サポート]({SUPPORT_URL})")
     gr.Markdown(f"[© コピーライト]({COPYRIGHT_URL})")
 
-# アプリを実行
 demo.launch()