Update app.py

app.py

@@ -1,4 +1,3 @@
-
 from youtube_transcript_api import (
     YouTubeTranscriptApi,
     NoTranscriptFound,
@@ -60,39 +59,48 @@ class YoutubeTranscript:
                 print(f"予期せぬエラーが発生しました: {str(e)}")
                 return None
 
-
-# The function 'get_transcript_for_gradio' should be outside the class
 def get_transcript_for_gradio(url: str, language: str) -> str:
-    """Gradio UI 用の書き起こし取得関数"""
     try:
         transcript = YoutubeTranscript.get_transcript(url, language)
         if transcript:
-            # formatted_transcript = "\n".join(
-            #     [f"[{entry['start']:.1f}s] {entry['text']}" for entry in transcript]
-            # )
             formatted_transcript = "".join(
                 [f"{entry['text']}" for entry in transcript]
             )
-            return formatted_transcript,str(len(formatted_transcript))
+            return formatted_transcript, str(len(formatted_transcript))
         else:
-            return "字幕の取得に失敗しました。","字幕の取得に失敗しました。"
+            return "字幕の取得に失敗しました。", "字幕の取得に失敗しました。"
     except NoTranscriptFound as e:
-        return str(e),str(len(e))  # Return the error message from the exception
+        return str(e), str(len(str(e)))
     except Exception as e:
-        return f"予期せぬエラーが発生しました: {str(e)}",str(len(e))
+        return f"予期せぬエラーが発生しました: {str(e)}", str(len(str(e)))
 
-iface = gr.Interface(
-    fn=get_transcript_for_gradio,
-    inputs=[
-        gr.Textbox(lines=1, placeholder="YouTube動画のURLを入力してください",show_copy_button=True),
-        gr.Radio(["ja", "en", "en-US"], label="言語", value="ja")  # ラジオボタンを追加
-    ],
-    outputs=[
-        gr.Code(label="字幕", max_lines=10),
-        gr.Textbox(label="文字数")  # Add a Textbox for character count
-    ],
-    title="YouTube字幕取得アプリ",
-    description="YouTube動画のURLを入力すると、字幕を取得して表示します。",
-)
+with gr.Blocks(title="YouTube字幕取得アプリ") as demo:
+    gr.Markdown("# YouTube字幕取得アプリ")
+    gr.Markdown("YouTube動画のURLを入力すると、字幕を取得して表示します。")
+
+    with gr.Row():
+        url_input = gr.Textbox(
+            lines=1,
+            placeholder="YouTube動画のURLを入力してください",
+            show_copy_button=True,
+            label="YouTube URL"
+        )
+        language_input = gr.Radio(
+            ["ja", "en", "en-US"],
+            label="言語",
+            value="ja"
+        )
+
+    submit_button = gr.Button("字幕を取得")
+
+    with gr.Column():
+        transcript_output = gr.Code(label="字幕", max_lines=10)
+        char_count_output = gr.Textbox(label="文字数")
+
+    submit_button.click(
+        fn=get_transcript_for_gradio,
+        inputs=[url_input, language_input],
+        outputs=[transcript_output, char_count_output]
+    )
 
-iface.launch()
+demo.launch()