Spaces:

MicroHealth
/

maiko-transcription

Paused

App Files Files Community

bluenevus commited on May 3

Commit

6dc9c9e

1 Parent(s): 4568c7d

Update app.py via AI Editor

Browse files

Files changed (1) hide show

app.py +155 -50

app.py CHANGED Viewed

@@ -7,18 +7,17 @@ import threading
 import logging
 import time
 import base64
-import json
 from datetime import datetime
 from flask import request
 import dash
 from dash import Dash, dcc, html, Input, Output, State, callback_context, no_update
 import dash_bootstrap_components as dbc
-import dash_daq as daq
 import requests
 import docx
 AUDIO_CHUNK_SIZE_MB = 25
-AUDIO_EXTS = ['.mp3', '.wav', '.m4a', '.ogg']
 TRANSCRIPT_FILENAME = 'transcript.docx'
 MINUTES_FILENAME = 'minutes.docx'
 RAW_AUDIO_FILENAME = 'meeting_audio.wav'
@@ -171,7 +170,108 @@ external_stylesheets = [dbc.themes.BOOTSTRAP, '/assets/custom.css']
 app = Dash(__name__, external_stylesheets=external_stylesheets, suppress_callback_exceptions=True)
 server = app.server
-def build_left_nav(sid, user_dir):
     audio_path = get_audio_path(user_dir)
     transcript_path = get_transcript_path(user_dir)
     minutes_path = get_minutes_path(user_dir)
@@ -186,22 +286,23 @@ def build_left_nav(sid, user_dir):
         links.append(dbc.NavLink("Meeting Minutes", href="#", id="nav-minutes", n_clicks=0, className="mb-2"))
         links.append(html.A("Download Minutes (docx)", href=f"/download/minutes/{sid}", target="_blank", id="dl-minutes", className="mb-2 d-block"))
     links.append(html.Hr())
-    links.append(dbc.Button("Delete All Files", id="btn-clear-all", color="secondary", className="mb-2", n_clicks=0))
     return dbc.Card(
         dbc.CardBody([
             html.H4("Navigation/Downloads", className="card-title"),
             html.Div(links, id="left-links"),
-            html.Hr(),
-            dbc.Button("Record Meeting", id="btn-record", color="primary", className="mb-2", n_clicks=0),
-            dcc.Upload(
-                id='upload-audio',
-                children=html.Div([
-                    'Or drag and drop/upload meeting audio here'
-                ]),
-                multiple=False,
-                accept="audio/*",
-                className='mb-2'
-            ),
         ]),
         className="mb-2"
     )
@@ -210,12 +311,12 @@ def build_right_preview(content_type, content, sid, user_dir):
     if content_type == "audio":
         audio_path = get_audio_path(user_dir)
         if not file_exists(audio_path):
-            return html.Div("No audio recorded/uploaded yet.")
         return html.Audio(src=f"/preview/audio/{sid}", controls=True, style={"width": "100%"})
     elif content_type == "transcript":
         transcript_path = get_transcript_path(user_dir)
         if not file_exists(transcript_path):
-            return html.Div("No transcript found. Please record/upload and transcribe meeting first.")
         with open(transcript_path, "rb") as f:
             doc = docx.Document(f)
             text = "\n".join([para.text for para in doc.paragraphs])
@@ -226,7 +327,7 @@ def build_right_preview(content_type, content, sid, user_dir):
     elif content_type == "minutes":
         minutes_path = get_minutes_path(user_dir)
         if not file_exists(minutes_path):
-            return html.Div("No meeting minutes found. Please transcribe meeting first.")
         with open(minutes_path, "rb") as f:
             doc = docx.Document(f)
             text = "\n".join([para.text for para in doc.paragraphs])
@@ -241,16 +342,8 @@ app.layout = dbc.Container([
     dcc.Store(id="store-session", storage_type="session"),
     dcc.Store(id="store-preview", data={"type": "none"}),
     dcc.Store(id="store-transcript-status", data={"status": "idle"}),
     dcc.Interval(id="interval-refresh", interval=60000, n_intervals=0),
-    # Hidden dcc.Upload to satisfy Dash callback registration
-    html.Div([
-        dcc.Upload(
-            id='upload-audio',
-            children=None,
-            style={"display": "none"}
-        )
-    ], style={"display": "none"}),
-    # Hidden dummy components for Dash callback recognition
     html.Div([
         dbc.Button(id="btn-record", style={"display": "none"}),
         dbc.Button(id="btn-clear-all", style={"display": "none"}),
@@ -292,20 +385,19 @@ app.layout = dbc.Container([
     Output("right-preview", "children"),
     Output("store-preview", "data"),
     Input("interval-refresh", "n_intervals"),
-    Input("upload-audio", "contents"),
-    Input("btn-record", "n_clicks"),
-    Input("store-transcript-status", "data"),
     Input("nav-audio", "n_clicks"),
     Input("nav-transcript", "n_clicks"),
     Input("nav-minutes", "n_clicks"),
     Input("btn-clear-all", "n_clicks"),
     State("store-session", "data"),
     State("store-preview", "data"),
     prevent_initial_call=False
 )
-def main_controller(interval_refresh, upload_contents, btn_record, transcript_status_data,
                     nav_audio, nav_transcript, nav_minutes, btn_clear_all,
-                    session_data, store_preview_data):
     ctx = callback_context
     triggered = ctx.triggered
@@ -314,7 +406,8 @@ def main_controller(interval_refresh, upload_contents, btn_record, transcript_st
     sid = session_manager.get_session_id()
     user_dir = session_manager.get_user_dir(sid)
     lock = session_manager.get_lock(sid)
-    left_nav = build_left_nav(sid, user_dir)
     right_preview = build_right_preview("none", None, sid, user_dir)
     store_preview = {"type": "none"}
     transcript_status = {"status": "idle"}
@@ -325,19 +418,18 @@ def main_controller(interval_refresh, upload_contents, btn_record, transcript_st
         transcript_status = transcript_status_data if transcript_status_data else {"status": "idle"}
         store_preview = store_preview_data if store_preview_data else {"type": "none"}
         right_preview = build_right_preview(store_preview.get('type', 'none'), None, sid, user_dir)
         return session_store, left_nav, transcript_status, right_preview, store_preview
-    elif trigger_id == "upload-audio" and upload_contents:
-        with lock:
-            save_uploaded_audio(upload_contents, user_dir)
-        logging.info(f"Audio uploaded for session: {sid}")
-        transcript_status = {"status": "audio_uploaded"}
-        left_nav = build_left_nav(sid, user_dir)
-        return session_store, left_nav, transcript_status, right_preview, store_preview
-    elif trigger_id == "btn-record":
-        transcript_status = {"status": "idle"}
-        return session_store, left_nav, transcript_status, right_preview, store_preview
     elif trigger_id == "store-transcript-status":
         if transcript_status_data and transcript_status_data.get("status") == "audio_uploaded":
@@ -356,16 +448,16 @@ def main_controller(interval_refresh, upload_contents, btn_record, transcript_st
                 minutes_path = get_minutes_path(user_dir)
                 write_docx(minutes_text, minutes_path)
                 logging.info(f"Transcription and minutes generated for session {sid}")
-            left_nav = build_left_nav(sid, user_dir)
             transcript_status = {"status": "done"}
             return session_store, left_nav, transcript_status, right_preview, store_preview
         else:
-            left_nav = build_left_nav(sid, user_dir)
             transcript_status = transcript_status_data if transcript_status_data else {"status": "idle"}
             return session_store, left_nav, transcript_status, right_preview, store_preview
     elif trigger_id in ["nav-audio", "nav-transcript", "nav-minutes"]:
-        left_nav = build_left_nav(sid, user_dir)
         if trigger_id == "nav-audio":
             right_preview = build_right_preview("audio", None, sid, user_dir)
             store_preview = {"type": "audio"}
@@ -382,18 +474,31 @@ def main_controller(interval_refresh, upload_contents, btn_record, transcript_st
         session_manager.clear_session(sid)
         logging.info(f"Cleared all files for session {sid}")
         user_dir = session_manager.get_user_dir(sid)
-        left_nav = build_left_nav(sid, user_dir)
         right_preview = build_right_preview("none", None, sid, user_dir)
         transcript_status = {"status": "idle"}
         store_preview = {"type": "none"}
         return session_store, left_nav, transcript_status, right_preview, store_preview
-    left_nav = build_left_nav(sid, user_dir)
     right_preview = build_right_preview("none", None, sid, user_dir)
     transcript_status = transcript_status_data if transcript_status_data else {"status": "idle"}
     store_preview = store_preview_data if store_preview_data else {"type": "none"}
     return session_store, left_nav, transcript_status, right_preview, store_preview
 @server.route('/preview/audio/<sid>')
 def serve_audio_preview(sid):
     user_dir = session_manager.get_user_dir(sid)

 import logging
 import time
 import base64
 from datetime import datetime
+import json
 from flask import request
 import dash
 from dash import Dash, dcc, html, Input, Output, State, callback_context, no_update
 import dash_bootstrap_components as dbc
 import requests
 import docx
 AUDIO_CHUNK_SIZE_MB = 25
+AUDIO_EXTS = ['.wav']
 TRANSCRIPT_FILENAME = 'transcript.docx'
 MINUTES_FILENAME = 'minutes.docx'
 RAW_AUDIO_FILENAME = 'meeting_audio.wav'
 app = Dash(__name__, external_stylesheets=external_stylesheets, suppress_callback_exceptions=True)
 server = app.server
+recording_js = '''
+window.dash_clientside = window.dash_clientside || {};
+window.dash_clientside.meetingrecorder = {
+    recorderState: {
+        mediaRecorder: null,
+        audioChunks: [],
+        isRecording: false
+    },
+    recordOrStop: function(nClicks, currentState) {
+        let that = window.dash_clientside.meetingrecorder;
+        if (!nClicks) {
+            return window.dash_clientside.no_update;
+        }
+        if (!that.recorderState.isRecording) {
+            navigator.mediaDevices.getUserMedia({ audio: { channelCount: 1 }, video: false }).then(function(stream) {
+                const mediaRecorder = new MediaRecorder(stream, { mimeType: 'audio/wav' });
+                that.recorderState.mediaRecorder = mediaRecorder;
+                that.recorderState.audioChunks = [];
+                that.recorderState.isRecording = true;
+                mediaRecorder.ondataavailable = function(event) {
+                    if (event.data.size > 0) {
+                        that.recorderState.audioChunks.push(event.data);
+                    }
+                };
+                mediaRecorder.onstop = function() {
+                    const audioBlob = new Blob(that.recorderState.audioChunks, {type: 'audio/wav'});
+                    const reader = new FileReader();
+                    reader.onloadend = function() {
+                        const base64data = reader.result;
+                        const payload = { audio: base64data };
+                        window.localStorage.setItem("meeting-audio-latest", base64data);
+                        const evt = new CustomEvent("meeting-audio-recorded");
+                        window.dispatchEvent(evt);
+                    };
+                    reader.readAsDataURL(audioBlob);
+                };
+                mediaRecorder.start();
+            });
+            return {isRecording: true};
+        } else {
+            if (that.recorderState.mediaRecorder) {
+                that.recorderState.mediaRecorder.stop();
+                that.recorderState.isRecording = false;
+            }
+            return {isRecording: false};
+        }
+    }
+};
+'''
+app.clientside_callback(
+    recording_js + '''
+    return window.dash_clientside.meetingrecorder.recordOrStop(nClicks, currentState);
+    ''',
+    Output("store-recording-state", "data"),
+    Input("btn-record", "n_clicks"),
+    State("store-recording-state", "data"),
+    prevent_initial_call=False
+)
+app.index_string = '''
+<!DOCTYPE html>
+<html>
+    <head>
+        {%metas%}
+        <title>Meeting Recorder & Transcriber</title>
+        {%favicon%}
+        {%css%}
+    </head>
+    <body>
+        {%app_entry%}
+        <script>
+        {recording_js}
+        document.addEventListener("DOMContentLoaded", function() {
+            if (!window.meetingAudioListenerRegistered) {
+                window.addEventListener("meeting-audio-recorded", function() {
+                    var audioData = window.localStorage.getItem("meeting-audio-latest");
+                    if (audioData) {
+                        var req = new XMLHttpRequest();
+                        req.open("POST", "/record_audio", true);
+                        req.setRequestHeader("Content-Type", "application/json;charset=UTF-8");
+                        req.onreadystatechange = function() {
+                            if (req.readyState === 4) {
+                                window.localStorage.removeItem("meeting-audio-latest");
+                                window.location.reload();
+                            }
+                        };
+                        req.send(JSON.stringify({ audio: audioData }));
+                    }
+                });
+                window.meetingAudioListenerRegistered = true;
+            }
+        });
+        </script>
+        {%config%}
+        {%scripts%}
+        {%renderer%}
+    </body>
+</html>
+'''.replace("{recording_js}", recording_js)
+def build_left_nav(sid, user_dir, is_recording):
     audio_path = get_audio_path(user_dir)
     transcript_path = get_transcript_path(user_dir)
     minutes_path = get_minutes_path(user_dir)
         links.append(dbc.NavLink("Meeting Minutes", href="#", id="nav-minutes", n_clicks=0, className="mb-2"))
         links.append(html.A("Download Minutes (docx)", href=f"/download/minutes/{sid}", target="_blank", id="dl-minutes", className="mb-2 d-block"))
     links.append(html.Hr())
+    links.append(dbc.Button(
+        "Delete All Files", id="btn-clear-all", color="secondary", className="mb-2", n_clicks=0))
+    record_label = "Stop Recording" if is_recording else "Record Meeting"
+    links.append(
+        dbc.Button(
+            record_label,
+            id="btn-record",
+            color="primary" if not is_recording else "danger",
+            className="mb-2",
+            n_clicks=0,
+            style={"width": "100%"}
+        )
+    )
     return dbc.Card(
         dbc.CardBody([
             html.H4("Navigation/Downloads", className="card-title"),
             html.Div(links, id="left-links"),
         ]),
         className="mb-2"
     )
     if content_type == "audio":
         audio_path = get_audio_path(user_dir)
         if not file_exists(audio_path):
+            return html.Div("No audio recorded yet.")
         return html.Audio(src=f"/preview/audio/{sid}", controls=True, style={"width": "100%"})
     elif content_type == "transcript":
         transcript_path = get_transcript_path(user_dir)
         if not file_exists(transcript_path):
+            return html.Div("No transcript found. Please record meeting first.")
         with open(transcript_path, "rb") as f:
             doc = docx.Document(f)
             text = "\n".join([para.text for para in doc.paragraphs])
     elif content_type == "minutes":
         minutes_path = get_minutes_path(user_dir)
         if not file_exists(minutes_path):
+            return html.Div("No meeting minutes found. Please record and transcribe meeting first.")
         with open(minutes_path, "rb") as f:
             doc = docx.Document(f)
             text = "\n".join([para.text for para in doc.paragraphs])
     dcc.Store(id="store-session", storage_type="session"),
     dcc.Store(id="store-preview", data={"type": "none"}),
     dcc.Store(id="store-transcript-status", data={"status": "idle"}),
+    dcc.Store(id="store-recording-state", data={"isRecording": False}),
     dcc.Interval(id="interval-refresh", interval=60000, n_intervals=0),
     html.Div([
         dbc.Button(id="btn-record", style={"display": "none"}),
         dbc.Button(id="btn-clear-all", style={"display": "none"}),
     Output("right-preview", "children"),
     Output("store-preview", "data"),
     Input("interval-refresh", "n_intervals"),
+    Input("store-recording-state", "data"),
     Input("nav-audio", "n_clicks"),
     Input("nav-transcript", "n_clicks"),
     Input("nav-minutes", "n_clicks"),
     Input("btn-clear-all", "n_clicks"),
     State("store-session", "data"),
     State("store-preview", "data"),
+    State("store-transcript-status", "data"),
     prevent_initial_call=False
 )
+def main_controller(interval_refresh, recording_state,
                     nav_audio, nav_transcript, nav_minutes, btn_clear_all,
+                    session_data, store_preview_data, transcript_status_data):
     ctx = callback_context
     triggered = ctx.triggered
     sid = session_manager.get_session_id()
     user_dir = session_manager.get_user_dir(sid)
     lock = session_manager.get_lock(sid)
+    is_recording = (recording_state or {}).get("isRecording", False)
+    left_nav = build_left_nav(sid, user_dir, is_recording)
     right_preview = build_right_preview("none", None, sid, user_dir)
     store_preview = {"type": "none"}
     transcript_status = {"status": "idle"}
         transcript_status = transcript_status_data if transcript_status_data else {"status": "idle"}
         store_preview = store_preview_data if store_preview_data else {"type": "none"}
         right_preview = build_right_preview(store_preview.get('type', 'none'), None, sid, user_dir)
+        left_nav = build_left_nav(sid, user_dir, is_recording)
         return session_store, left_nav, transcript_status, right_preview, store_preview
+    elif trigger_id == "store-recording-state":
+        audio_path = get_audio_path(user_dir)
+        if file_exists(audio_path) and not is_recording:
+            left_nav = build_left_nav(sid, user_dir, False)
+            transcript_status = {"status": "audio_uploaded"}
+            return session_store, left_nav, transcript_status, right_preview, store_preview
+        else:
+            left_nav = build_left_nav(sid, user_dir, is_recording)
+            return session_store, left_nav, transcript_status_data if transcript_status_data else {"status": "idle"}, right_preview, store_preview
     elif trigger_id == "store-transcript-status":
         if transcript_status_data and transcript_status_data.get("status") == "audio_uploaded":
                 minutes_path = get_minutes_path(user_dir)
                 write_docx(minutes_text, minutes_path)
                 logging.info(f"Transcription and minutes generated for session {sid}")
+            left_nav = build_left_nav(sid, user_dir, False)
             transcript_status = {"status": "done"}
             return session_store, left_nav, transcript_status, right_preview, store_preview
         else:
+            left_nav = build_left_nav(sid, user_dir, is_recording)
             transcript_status = transcript_status_data if transcript_status_data else {"status": "idle"}
             return session_store, left_nav, transcript_status, right_preview, store_preview
     elif trigger_id in ["nav-audio", "nav-transcript", "nav-minutes"]:
+        left_nav = build_left_nav(sid, user_dir, is_recording)
         if trigger_id == "nav-audio":
             right_preview = build_right_preview("audio", None, sid, user_dir)
             store_preview = {"type": "audio"}
         session_manager.clear_session(sid)
         logging.info(f"Cleared all files for session {sid}")
         user_dir = session_manager.get_user_dir(sid)
+        left_nav = build_left_nav(sid, user_dir, False)
         right_preview = build_right_preview("none", None, sid, user_dir)
         transcript_status = {"status": "idle"}
         store_preview = {"type": "none"}
         return session_store, left_nav, transcript_status, right_preview, store_preview
+    left_nav = build_left_nav(sid, user_dir, is_recording)
     right_preview = build_right_preview("none", None, sid, user_dir)
     transcript_status = transcript_status_data if transcript_status_data else {"status": "idle"}
     store_preview = store_preview_data if store_preview_data else {"type": "none"}
     return session_store, left_nav, transcript_status, right_preview, store_preview
+@server.route('/record_audio', methods=["POST"])
+def record_audio():
+    sid = session_manager.get_session_id()
+    user_dir = session_manager.get_user_dir(sid)
+    data = request.get_json(force=True)
+    audio_b64 = data.get("audio", "")
+    if audio_b64:
+        save_uploaded_audio(audio_b64, user_dir)
+        logging.info(f"Audio recorded and saved for session {sid}")
+        return "OK", 200
+    else:
+        return "No audio data", 400
 @server.route('/preview/audio/<sid>')
 def serve_audio_preview(sid):
     user_dir = session_manager.get_user_dir(sid)