Spaces:
Runtime error
Runtime error
| 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) | |