Update app.py

app.py

@@ -46,134 +46,50 @@ if not OPENAI_API_KEY:
 
 openai.api_key = OPENAI_API_KEY
 
-def is_valid_url(url):
-    try:
-        result = urlparse(url)
-        return all([result.scheme, result.netloc])
-    except ValueError:
-        logger.error(f"Invalid URL: {url}")
-        return False
-
-def download_media(url):
-    logger.info(f"Attempting to download media from URL: {url}")
-    try:
-        if "youtube.com" in url or "youtu.be" in url:
-            yt = YouTube(url)
-            stream = yt.streams.filter(progressive=True, file_extension='mp4').first()
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_file:
-                stream.download(output_path=os.path.dirname(temp_file.name), filename=temp_file.name)
-                logger.info(f"YouTube video downloaded: {temp_file.name}")
-                return temp_file.name
-        else:
-            response = requests.get(url)
-            content_type = response.headers.get('content-type', '')
-            if 'video' in content_type:
-                suffix = '.mp4'
-            elif 'audio' in content_type:
-                suffix = '.mp3'
-            else:
-                suffix = ''
-            with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
-                temp_file.write(response.content)
-                logger.info(f"Media downloaded: {temp_file.name}")
-                return temp_file.name
-    except Exception as e:
-        logger.error(f"Error downloading media: {str(e)}")
-        raise
-
-def extract_audio(file_path):
-    logger.info(f"Extracting audio from video: {file_path}")
-    try:
-        if VideoFileClip is None:
-            raise ImportError("VideoFileClip is not available. Cannot extract audio.")
-        video = VideoFileClip(file_path)
-        audio = video.audio
-        audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
-        audio.write_audiofile(audio_file.name)
-        video.close()
-        audio.close()
-        logger.info(f"Audio extracted: {audio_file.name}")
-        return audio_file.name
-    except Exception as e:
-        logger.error(f"Error extracting audio: {str(e)}")
-        raise
-
-def transcribe_audio(file_path):
-    logger.info(f"Transcribing audio: {file_path}")
-    try:
-        with open(file_path, "rb") as audio_file:
-            transcript = openai.Audio.transcribe("whisper-1", audio_file)
-        logger.info("Transcription completed successfully")
-        return transcript["text"]
-    except Exception as e:
-        logger.error(f"Error during transcription: {str(e)}")
-        raise
-
-def process_media(contents, filename, url):
-    logger.info("Starting media processing")
-    try:
-        if contents:
-            content_type, content_string = contents.split(',')
-            decoded = base64.b64decode(content_string)
-            suffix = os.path.splitext(filename)[1]
-            with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
-                temp_file.write(decoded)
-                temp_file_path = temp_file.name
-            logger.info(f"File uploaded: {temp_file_path}")
-        elif url:
-            temp_file_path = download_media(url)
-        else:
-            logger.error("No input provided")
-            raise ValueError("No input provided")
-
-        if temp_file_path.lower().endswith(('.mp4', '.avi', '.mov', '.flv', '.wmv')):
-            logger.info("Video file detected, extracting audio")
-            audio_file_path = extract_audio(temp_file_path)
-            transcript = transcribe_audio(audio_file_path)
-            os.unlink(audio_file_path)
-        else:
-            logger.info("Audio file detected, transcribing directly")
-            transcript = transcribe_audio(temp_file_path)
-
-        os.unlink(temp_file_path)
-        return transcript
-    except Exception as e:
-        logger.error(f"Error in process_media: {str(e)}")
-        raise
+# ... (keep all the helper functions as they are) ...
 
 app.layout = dbc.Container([
-    html.H1("Audio/Video Transcription App", className="text-center my-4"),
-    dbc.Card([
-        dbc.CardBody([
-            dcc.Upload(
-                id='upload-media',
-                children=html.Div([
-                    'Drag and Drop or ',
-                    html.A('Select Audio/Video File')
-                ]),
-                style={
-                    'width': '100%',
-                    'height': '60px',
-                    'lineHeight': '60px',
-                    'borderWidth': '1px',
-                    'borderStyle': 'dashed',
-                    'borderRadius': '5px',
-                    'textAlign': 'center',
-                    'margin': '10px'
-                },
-                multiple=False
-            ),
-            html.Div(id='file-info', className="mt-2"),
-            dbc.Input(id="media-url", type="text", placeholder="Enter audio/video URL or YouTube link", className="my-3"),
-            dbc.Button("Transcribe", id="transcribe-button", color="primary", className="w-100 mb-3"),
-            dbc.Spinner(html.Div(id="transcription-output", className="mt-3")),
-            html.Div(id="progress-indicator", className="text-center mt-3"),
-            dbc.Button("Download Transcript", id="download-button", color="secondary", className="w-100 mt-3", style={'display': 'none'}),
-            dcc.Download(id="download-transcript"),
-            dcc.Interval(id='progress-interval', interval=500, n_intervals=0, disabled=True)
+    dbc.Row([
+        dbc.Col([
+            html.H1("Audio/Video Transcription App", className="text-center my-4"),
         ])
+    ]),
+    dbc.Row([
+        dbc.Col([
+            dbc.Card([
+                dbc.CardBody([
+                    dcc.Upload(
+                        id='upload-media',
+                        children=html.Div([
+                            'Drag and Drop or ',
+                            html.A('Select Audio/Video File')
+                        ]),
+                        style={
+                            'width': '100%',
+                            'height': '60px',
+                            'lineHeight': '60px',
+                            'borderWidth': '1px',
+                            'borderStyle': 'dashed',
+                            'borderRadius': '5px',
+                            'textAlign': 'center',
+                            'margin': '10px'
+                        },
+                        multiple=False
+                    ),
+                    html.Div(id='file-info', className="mt-2"),
+                    dbc.Input(id="media-url", type="text", placeholder="Enter audio/video URL or YouTube link", className="my-3"),
+                    dbc.Button("Transcribe", id="transcribe-button", color="primary", className="w-100 mb-3"),
+                    dbc.Spinner(html.Div(id="transcription-output", className="mt-3")),
+                    html.Div(id="progress-indicator", className="text-center mt-3"),
+                    dbc.Button("Download Transcript", id="download-button", color="secondary", className="w-100 mt-3", style={'display': 'none'}),
+                    dcc.Download(id="download-transcript"),
+                    dcc.Store(id="transcription-store"),
+                    dcc.Interval(id='progress-interval', interval=500, n_intervals=0, disabled=True)
+                ])
+            ])
+        ], width=12)
     ])
-])
+], fluid=True)
 
 @app.callback(
     Output("file-info", "children"),
@@ -190,14 +106,16 @@ def update_file_info(filename, last_modified):
     Output("download-button", "style"),
     Output("progress-indicator", "children"),
     Output("progress-interval", "disabled"),
+    Output("transcription-store", "data"),
     Input("transcribe-button", "n_clicks"),
     Input("progress-interval", "n_intervals"),
     State("upload-media", "contents"),
     State("upload-media", "filename"),
     State("media-url", "value"),
+    State("transcription-store", "data"),
     prevent_initial_call=True
 )
-def update_transcription(n_clicks, n_intervals, contents, filename, url):
+def update_transcription(n_clicks, n_intervals, contents, filename, url, stored_transcript):
     ctx = callback_context
     if ctx.triggered_id == "transcribe-button":
         if not contents and not url:
@@ -212,40 +130,43 @@ def update_transcription(n_clicks, n_intervals, contents, filename, url):
 
         thread = threading.Thread(target=transcribe)
         thread.start()
-        return html.Div("Processing..."), {'display': 'none'}, "", False
+        return html.Div("Processing..."), {'display': 'none'}, "", False, None
 
     elif ctx.triggered_id == "progress-interval":
+        if stored_transcript:
+            return display_transcript(stored_transcript), {'display': 'block'}, "", True, stored_transcript
         dots = "." * (n_intervals % 4)
-        return html.Div("Processing" + dots), {'display': 'none'}, "", False
+        return html.Div("Processing" + dots), {'display': 'none'}, "", False, None
 
     thread = threading.current_thread()
     if hasattr(thread, 'result'):
         transcript = thread.result
         if transcript and not transcript.startswith("An error occurred"):
             logger.info("Transcription successful")
-            return dbc.Card([
-                dbc.CardBody([
-                    html.H5("Transcription Result"),
-                    html.Pre(transcript, style={"white-space": "pre-wrap", "word-wrap": "break-word"})
-                ])
-            ]), {'display': 'block'}, "", True
+            return display_transcript(transcript), {'display': 'block'}, "", True, transcript
         else:
             logger.error(f"Transcription failed: {transcript}")
-            return transcript, {'display': 'none'}, "", True
+            return html.Div(transcript), {'display': 'none'}, "", True, None
 
-    return dash.no_update, dash.no_update, dash.no_update, dash.no_update
+    return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update
+
+def display_transcript(transcript):
+    return dbc.Card([
+        dbc.CardBody([
+            html.H5("Transcription Result"),
+            html.Pre(transcript, style={"white-space": "pre-wrap", "word-wrap": "break-word"})
+        ])
+    ])
 
 @app.callback(
     Output("download-transcript", "data"),
     Input("download-button", "n_clicks"),
-    State("transcription-output", "children"),
+    State("transcription-store", "data"),
     prevent_initial_call=True
 )
-def download_transcript(n_clicks, transcription_output):
-    if not transcription_output:
+def download_transcript(n_clicks, transcript):
+    if not transcript:
         raise PreventUpdate
-    
-    transcript = transcription_output['props']['children'][0]['props']['children'][1]['props']['children']
     return dict(content=transcript, filename="transcript.txt")
 
 if __name__ == '__main__':