Update app.py

app.py

@@ -43,7 +43,6 @@ transcription_text = ""
 # Set up OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-# Layout
 app.layout = dbc.Container([
     html.H1("Audio/Video Transcription and Diarization App", className="text-center my-4"),
     dbc.Card([
@@ -73,8 +72,12 @@ app.layout = dbc.Container([
             html.H4("Diarized Transcription Preview", className="mt-4"),
             html.Div(id='transcription-preview', style={'whiteSpace': 'pre-wrap'}),
             html.Br(),
-            dbc.Button("Download Transcription", id="btn-download", color="primary", className="mt-3", disabled=True),
-            dcc.Download(id="download-transcription")
+            dbc.Button("Download Transcription", id="btn-download", color="primary", className="mt-3 me-2", disabled=True),
+            dbc.Button("Summarize Transcript", id="btn-summarize", color="secondary", className="mt-3 me-2", disabled=True),
+            dbc.Button("Generate Meeting Minutes", id="btn-minutes", color="info", className="mt-3", disabled=True),
+            dcc.Download(id="download-transcription"),
+            html.Div(id='summary-output', className="mt-4"),
+            html.Div(id='minutes-output', className="mt-4")
         ])
     ])
 ], fluid=True)
@@ -243,12 +246,94 @@ def process_media(file_path, is_url=False):
             os.unlink(temp_file)
         if wav_path and os.path.exists(wav_path):
             os.unlink(wav_path)
-            
+
+@app.callback(
+    Output('summary-output', 'children'),
+    Input('btn-summarize', 'n_clicks'),
+    State('transcription-preview', 'children'),
+    prevent_initial_call=True
+)
+def summarize_transcript(n_clicks, transcript):
+    if n_clicks is None or not transcript:
+        return ""
+    
+    summary_prompt = f"""
+    Please provide a concise summary of the following transcript. Include the main topics discussed and key points. Format it for readability:
+
+    {transcript}
+
+    Summary:
+    """
+
+    try:
+        summary_response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": "You are an AI assistant skilled in summarizing conversations."},
+                {"role": "user", "content": summary_prompt}
+            ]
+        )
+
+        summary = summary_response['choices'][0]['message']['content']
+        return dbc.Card(dbc.CardBody([
+            html.H5("Transcript Summary", className="card-title"),
+            html.P(summary, className="card-text")
+        ]))
+    except Exception as e:
+        logger.error(f"Error generating summary: {str(e)}")
+        return html.Div(f"An error occurred while generating the summary: {str(e)}", className="text-danger")
+
+@app.callback(
+    Output('minutes-output', 'children'),
+    Input('btn-minutes', 'n_clicks'),
+    State('transcription-preview', 'children'),
+    prevent_initial_call=True
+)
+def generate_meeting_minutes(n_clicks, transcript):
+    if n_clicks is None or not transcript:
+        return ""
+    
+    minutes_prompt = f"""
+    Please transform the following transcript into structured meeting minutes. Include the following sections:
+    1. Meeting Title
+    2. Date and Time (if mentioned)
+    3. Attendees (if mentioned)
+    4. Agenda Items
+    5. Key Decisions
+    6. Action Items
+    7. Next Steps
+
+    Transcript:
+    {transcript}
+
+    Meeting Minutes:
+    """
+
+    try:
+        minutes_response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system", "content": "You are an AI assistant skilled in creating structured meeting minutes from transcripts."},
+                {"role": "user", "content": minutes_prompt}
+            ]
+        )
+
+        minutes = minutes_response['choices'][0]['message']['content']
+        return dbc.Card(dbc.CardBody([
+            html.H5("Meeting Minutes", className="card-title"),
+            dcc.Markdown(minutes, className="card-text")
+        ]))
+    except Exception as e:
+        logger.error(f"Error generating meeting minutes: {str(e)}")
+        return html.Div(f"An error occurred while generating meeting minutes: {str(e)}", className="text-danger")
+
 @app.callback(
     [Output('output-media-upload', 'children'),
      Output('transcription-status', 'children'),
      Output('transcription-preview', 'children'),
-     Output('btn-download', 'disabled')],
+     Output('btn-download', 'disabled'),
+     Output('btn-summarize', 'disabled'),
+     Output('btn-minutes', 'disabled')],
     [Input('upload-media', 'contents'),
      Input('process-url-button', 'n_clicks')],
     [State('upload-media', 'filename'),
@@ -257,7 +342,7 @@ def process_media(file_path, is_url=False):
 def update_output(contents, n_clicks, filename, url):
     ctx = callback_context
     if not ctx.triggered:
-        return "No file uploaded or URL processed.", "", "", True
+        return "No file uploaded or URL processed.", "", "", True, True, True
 
     # Clear the preview pane
     transcription_preview = ""
@@ -267,13 +352,13 @@ def update_output(contents, n_clicks, filename, url):
     elif url:
         status_message, success = process_media(url, is_url=True)
     else:
-        return "No file uploaded or URL processed.", "", "", True
+        return "No file uploaded or URL processed.", "", "", True, True, True
 
     if success:
         preview = transcription_text[:1000] + "..." if len(transcription_text) > 1000 else transcription_text
-        return f"Media processed successfully.", status_message, preview, False
+        return f"Media processed successfully.", status_message, preview, False, False, False
     else:
-        return "Processing failed.", status_message, transcription_preview, True
+        return "Processing failed.", status_message, transcription_preview, True, True, True
 
 @app.callback(
     Output("download-transcription", "data"),