upload

app.py

@@ -1,10 +1,35 @@
+import os
+import json
+import base64
+
 import gradio as gr
+import firebase_admin
+from firebase_admin import credentials
+from firebase_admin import firestore
 
 from utils import get_youtube
 from model import summarize_video
 
 root_dir = '/home/user/app/video'
 
+db = firestore.client()
+# 環境変数から秘密鍵を取得
+encoded_key = os.environ["FIREBASE_CREDENTIALS_BASE64"]
+# Base64エンコードされた秘密鍵をデコード
+decoded_key = base64.b64decode(encoded_key)
+# デコードされた秘密鍵を使ってCredentialオブジェクトを作成
+cred = credentials.Certificate(json.loads(decoded_key))
+# Firebase Admin SDKを初期化
+firebase_admin.initialize_app(cred)
+
+def log_firestore(user_id="000000", message="test"):
+    doc_ref = db.collection("button_clicks").document()
+    doc_ref.set({
+        "user_id": user_id,
+        "message": message,
+        "timestamp": firestore.SERVER_TIMESTAMP
+    })
+
 # ---- Gradio Layout -----
 youtube_url_in = gr.Textbox(label="Youtube url", lines=1, interactive=True)
 video_in = gr.Video(label="Input Video", mirror_webcam=False, interactive=True)
@@ -26,10 +51,15 @@ with demo:
             ### Summarize video
             <ul>
                 <li>Step 1. download a video from youtube (select one of the examples and press the Download button)</li>
-                <li>Step 2: Select the summary rate and playback speed</li>
+                <li>Step 2: Select the summary rate and playback speed (select with slider)</li>
                 <li>Step 3: Generate a summarized video (press the Summarize button)</li>
             </ul>
-            A summarized video will be generated on the right side of the original video. In addition, the summarized text of the video and in the video
+            #### Output
+            <ul>
+                <li>An original downloaded video is displayed on the left side of the screen, and a summarized video is displayed on the right side of the screen.</li>
+                <li>A text summary of the video is displayed below it.</li>
+                <li>A transcription of the original video is displayed further down (the part extracted from the summarized video is highlighted).</li>
+            </ul>
             ''')
     with gr.Row():
             gr.Markdown('''

model.py

@@ -13,7 +13,8 @@ from transformers import pipeline, BertTokenizer, BertForNextSentencePrediction
 import torch 
 import whisper
 
-from utils import two_chnnel_to_one_channel, convert_sample_rate, log_firestore
+from utils import two_chnnel_to_one_channel, convert_sample_rate
+from app import log_firestore
 
 subprocess.run(['apt-get', '-y', 'install', 'imagemagick'])
 

utils.py

@@ -1,30 +1,7 @@
-import os
-import json
-import base64
-
-import firebase_admin
-from firebase_admin import credentials
-from firebase_admin import firestore
 from pytube import YouTube
 from scipy.signal import resample
 
-db = firestore.client()
-# 環境変数から秘密鍵を取得
-encoded_key = os.environ["FIREBASE_CREDENTIALS_BASE64"]
-# Base64エンコードされた秘密鍵をデコード
-decoded_key = base64.b64decode(encoded_key)
-# デコードされた秘密鍵を使ってCredentialオブジェクトを作成
-cred = credentials.Certificate(json.loads(decoded_key))
-# Firebase Admin SDKを初期化
-firebase_admin.initialize_app(cred)
-
-def log_firestore(user_id="000000", message="test"):
-    doc_ref = db.collection("button_clicks").document()
-    doc_ref.set({
-        "user_id": user_id,
-        "message": message,
-        "timestamp": firestore.SERVER_TIMESTAMP
-    })
+from app import log_firestore
 
 def get_youtube(user_id, video_url):
     # YouTubeの動画をダウンロード