add app

app.py

@@ -0,0 +1,76 @@
+import base64
+import os
+import time
+
+import banana_dev as banana
+import gradio as gr
+from loguru import logger
+
+api_key = os.environ["BANANA_API_KEY"]
+model_key = os.environ["BANANA_MODEL_KEY"]
+password = os.environ["PASSWORD"]
+
+
+def transcribe(email: str, file=None, url=None):
+    if file:
+        with open(file, "rb") as f:
+            audio = f.read()
+            audio_b64 = base64.b64encode(audio).decode("ISO-8859-1")
+        payload = {"audio_b64": audio_b64}
+    else:
+        payload = {"url": url}
+    payload["email"] = email
+    response = banana.run(api_key, model_key, payload)
+    print(response)
+
+    if "error" in response:
+        raise gr.Error(response["error"])
+
+    # TODO: not sure why response dict contains multiple model outputs
+    return response["modelOutputs"][0]
+
+
+def run_demo(email, password, microphone, file_upload):
+    if password != os.environ["PASSWORD"]:
+        raise gr.Error("Das Kennwort ist falsch.")
+
+    if (microphone is not None) and (file_upload is not None):
+        logger.warning(
+            "Achtung: Sie haben sowohl eine Datei hochgeladen als auch über das Mikrofon aufgenommen."
+            " Wir verwenden nur die Datei, die Sie hochgeladen haben."
+        )
+
+    elif (microphone is None) and (file_upload is None):
+        raise gr.Error(
+            "Sie müssen entweder eine Datei hochladen oder über das Mikrofon aufnehmen."
+        )
+
+    file = microphone if microphone is not None else file_upload
+
+    start = time.time()
+    transcription = transcribe(email, file)
+    logger.info(f"transcription took {time.time()-start:.3f}s")
+    return "\n\n".join([seg["text"].strip() for seg in transcription["segments"]])
+
+
+demo = gr.Interface(
+    fn=run_demo,
+    inputs=[
+        gr.Textbox(label="Email", type="email"),
+        gr.Textbox(label="Zugriffs-Kennwort (siehe oben)"),
+        gr.Audio(source="microphone", type="filepath", label="Aufnehmen"),
+        gr.Audio(source="upload", type="filepath", label="Datei hochladen"),
+    ],
+    outputs=gr.Textbox(label="Transkription"),
+    title="Transkriptionsservice",
+    description="<p>Die untenstehende, automatische Transkription wurde von uns für interne Zwecke entwickelt."
+    " Sie liefert die bei weitem beste Transkriptionsqualität bei Aufnahmen nicht-fachspezifischer Gespräche.</p>"
+    "<p><strong>Bitte geben Sie Ihre Email-Adresse an, damit wir Ihnen die fertige Transkription per Email zusenden können.</strong>"
+    " Wir werden Ihre Email-Adresse nicht für andere Zwecke verwenden.</p>"
+    "",
+    article="",
+    allow_flagging="never",
+    css="footer {visibility: hidden}",
+)
+
+demo.launch(share=True)

requirements.txt

@@ -0,0 +1,3 @@
+banana-dev
+gradio
+loguru