Update app.py

app.py

@@ -7,7 +7,7 @@ import tempfile
 import base64
 from concurrent.futures import ThreadPoolExecutor
 import logging
-import time  # インポート追加
+import time
 
 # ロギング設定
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -98,7 +98,7 @@ def process_pdf(pdf_file, progress=gr.Progress()):
         # PDFを分割
         split_pdf_paths = split_pdf(temp_pdf_path, temp_dir)
         logging.info(f"Split PDF paths: {split_pdf_paths}")
-        progress(0.2, desc="PDFを分割中...") # 進捗更新
+        progress(0.2, desc="PDFを分割中...")  # 進捗更新
 
         # 並列処理でOCR変換
         markdown_results = []
@@ -109,17 +109,17 @@ def process_pdf(pdf_file, progress=gr.Progress()):
                 try:
                     result = future.result()
                     markdown_results.append(result)
-                    progress(0.2 + 0.6 * (i + 1) / len(futures), desc="OCR処理中...") # 進捗更新
+                    progress(0.2 + 0.6 * (i + 1) / len(futures), desc="OCR処理中...")  # 進捗更新
                 except Exception as e:
                     logging.error(f"Error processing split PDF: {e}")
                     markdown_results.append(f"分割PDFの処理中にエラーが発生しました: {e}")
 
         logging.info(f"Markdown results length: {len(markdown_results)}")
-        progress(0.8, desc="結果を結合中...")# 進捗更新
+        progress(0.8, desc="結果を結合中...")  # 進捗更新
         # 結果を結合
         combined_markdown = "\n\n".join(markdown_results)
-        progress(1.0, desc="完了") # 進捗更新
-        time.sleep(0.5) #完了表示のため少し待つ
+        progress(1.0, desc="完了")  # 進捗更新
+        time.sleep(0.5)  # 完了表示のため少し待つ
 
         return combined_markdown
 
@@ -134,21 +134,31 @@ def create_interface():
             pdf_input = gr.File(label="PDFファイルをアップロード", file_types=[".pdf"])
 
         with gr.Row():
-            convert_btn = gr.Button("変換開始", variant="primary", elem_id="convert-button") # variantとelem_idを追加
+            convert_btn = gr.Button("変換開始", variant="primary",
+                                    elem_id="convert-button")  # variantとelem_idを追加
 
         with gr.Row():
-            markdown_output = gr.Textbox(label="変換結果", lines=10, max_lines=20) # MarkdownからTextboxに変更、行数を指定
+            markdown_output = gr.Textbox(label="変換結果", lines=10,
+                                         max_lines=20)  # MarkdownからTextboxに変更、行数を指定
 
         with gr.Row():
-          copy_btn = gr.Button("クリップボードにコピー")
-          download_btn = gr.Button("ダウンロード")
+            copy_btn = gr.Button("クリップボードにコピー")
+            download_btn = gr.Button("ダウンロード")
+
+        # JavaScriptコードを埋め込むためのHTMLコンポーネント
+        js_code = gr.HTML(
+            """
+            <script>
+            function styleButton() {
+              document.getElementById('convert-button').style.backgroundColor = 'orange';
+            }
+            </script>
+            """,
+            visible=False, # 非表示にしておく
+        )
 
-        # スタイル設定 (CSS)
-        demo.load(None, None, None, _js="""
-        () => {
-          document.getElementById('convert-button').style.backgroundColor = 'orange';
-        }
-        """)
+        # js_code が読み込まれた後に styleButton 関数を実行
+        pdf_input.upload(None, [], [], js="styleButton")
 
 
         convert_btn.click(
@@ -167,10 +177,10 @@ def create_interface():
 
         # ダウンロード
         download_btn.click(
-          None,
-          markdown_output,
-          [],
-          js=f"""(x) =>{{
+            None,
+            markdown_output,
+            [],
+            js=f"""(x) =>{{
             const blob = new Blob([x], {{type: 'text/markdown;charset=utf-8'}});
             const url = URL.createObjectURL(blob);
             const a = document.createElement('a');