Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import re
 import os
+import base64
 from datetime import datetime, timedelta, timezone
 
 # 設定
@@ -76,28 +77,25 @@ def extract_date_range_logs(lines, start_date, end_date):
 
     return results
 
-def export_results(results, search_condition):
-    """検索結果をTXTファイルとしてエクスポート"""
+def generate_download_link(results, search_condition):
+    """検索結果をTXTファイルとしてbase64エンコードし、ダウンロードリンクを生成"""
     if not results:
-        return None, "エラー: エクスポートするデータがありません。"
-
-    filename = f"search_results_{datetime.now(JST).strftime('%Y%m%d_%H%M%S')}.txt"
-    file_path = os.path.join("/tmp", filename)
+        return "エラー: エクスポートするデータがありません。"
 
     execution_date = datetime.now(JST).strftime("%Y/%m/%d %H:%M:%S")
 
-    try:
-        with open(file_path, "w", encoding="utf-8") as f:
-            f.write("プログラム名: LINEログマネージャー\n")
-            f.write(f"検索実行日: {execution_date}\n")
-            f.write(f"{search_condition}\n")
-            f.write(f"ヒット件数: {len(results.splitlines())} 件\n")
-            f.write("-" * 40 + "\n")
-            f.write(results + "\n")
-
-        return file_path, "エクスポート完了"
-    except Exception as e:
-        return None, f"エラー: {str(e)}"
+    file_content = (
+        "プログラム名: LINEログマネージャー\n"
+        f"検索実行日: {execution_date}\n"
+        f"{search_condition}\n"
+        f"ヒット件数: {len(results.splitlines())} 件\n"
+        + "-" * 40 + "\n"
+        + results + "\n"
+    )
+
+    encoded_file = base64.b64encode(file_content.encode()).decode()
+    href = f'<a href="data:text/plain;base64,{encoded_file}" download="search_results.txt">結果をダウンロード</a>'
+    return href
 
 # Gradio UI
 with gr.Blocks() as demo:
@@ -121,7 +119,7 @@ with gr.Blocks() as demo:
 
     # エクスポート機能
     export_button = gr.Button("TXTをエクスポートする")
-    export_output = gr.File(label="結果をダウンロード", interactive=True)  # "ファイル名.txt" ではなく固定
+    export_output = gr.HTML()  # base64エンコードしたリンクを表示
     export_status = gr.Textbox(label="エクスポートメッセージ", interactive=False)
 
     search_button.click(
@@ -131,9 +129,9 @@ with gr.Blocks() as demo:
     )
 
     export_button.click(
-        export_results, 
+        generate_download_link, 
         inputs=[result_table, search_condition_state], 
-        outputs=[export_output, export_status]
+        outputs=[export_output]
     )
 
     gr.Markdown(f"[📖 ヘルプページ]({SUPPORT_URL})")