Update app.py

app.py

@@ -3,6 +3,7 @@ import json
 import os
 
 import modal
+import requests
 import streamlit as st
 from loguru import logger
 from pydub import AudioSegment
@@ -17,8 +18,13 @@ run_transcription = modal.Function.lookup(
 st.set_page_config(page_title="Speech to Text Transcription App")
 
 
+@st.cache_data(show_spinner=False)
 def transcribe(url, audio_b64, cutoff):
-    return run_transcription.remote(url=url, audio_b64=audio_b64, cutoff=cutoff, language=None)
+    payload = {"url": url, "audio_b64": audio_b64, "cutoff": cutoff}
+    response = requests.post(
+        "https://aseifert--ffpub-transcription-fastapi-app.modal.run/transcribe", json=payload
+    )
+    return response.json()
 
 
 def password_is_correct(password):
@@ -61,6 +67,10 @@ def run():
             "URL (e.g. YouTube video, Dropbox file, etc.)",
             value="",
         )
+        if "youtu" in url:
+            url = url.replace("youtu.be/", "youtube.com/watch?v=")
+        elif "dropbox" in url:
+            url = url.replace("dl=0", "raw=1")
     else:
         audio_file = col2.file_uploader(
             "Datei auswählen", type=[".wav", ".mp3", ".flac", ".m4a", ".ogg"]
@@ -73,14 +83,16 @@ def run():
     )
 
     cutoff = audio_b64 = None
-    cutoff = 60_000_000 if password in ROOT_PASSWORDS else 60_000
+    cutoff = None if password in ROOT_PASSWORDS else 60
     if audio_file or url:
         # with st.expander(("Audio" if audio_file else "Video") + " abspielen"):
         if audio_file:
             st.audio(audio_file)
-            audio_file = AudioSegment.from_file(audio_file)[:cutoff]
+            audio_file = AudioSegment.from_file(audio_file)[: cutoff * 1000 if cutoff else None]
             audio_b64 = base64.b64encode(audio_file.export().read()).decode("ascii")
         if url:
+            if url == "https://www.youtube.com/watch?v=6UONiGMmbS4":
+                cutoff = None
             st.video(url)
 
     if input_is_ready(password, audio_file, url) and submit_button: