Update app.py

first attempt at combining save-transcribe-fix single click

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 from pydub import AudioSegment
 
 # Initialize the Groq client
-client = Groq(api_key=os.environ["keko"])
+client = Groq(api_key=os.environ["keko"]) 
 
 
 
@@ -284,10 +284,26 @@ def transcribe_and_generate(audio_file, prompt_type):
     
     return transcription, generated_text
 
+# Define a new function that combines save, transcribe, and fix
+def save_transcribe_fix(audio_file, prompt_type):
+    # Step 1: Save the audio
+    saved_audio, save_status = save_audio(audio_file)
+    if not saved_audio:
+        return None, save_status, None  # Return error message if saving fails
+
+    # Step 2: Transcribe the audio
+    transcription = transcribe_audio(saved_audio)
+    
+    # Step 3: Generate the fixed text
+    fixed_text = generate_text(transcription, prompt_type)
+    
+    return saved_audio, transcription, fixed_text
+
 
 # ------------------------------------------------
 # Create the Gradio interface using Blocks
 # -------------------------------------------------
+# Update Gradio UI with a new button
 with gr.Blocks(title="Audio Recorder, Transcriber, and Text Generator") as demo:
     gr.Markdown("# Audio Recorder, Transcriber, and Text Generator")
     gr.Markdown("Enter the correct password to access the application.")
@@ -305,7 +321,8 @@ with gr.Blocks(title="Audio Recorder, Transcriber, and Text Generator") as demo:
         with gr.Row():
             save_btn = gr.Button("Save Audio")
             transcribe_btn = gr.Button("Transcribe Audio")
-            transcribe_fix_btn = gr.Button("Transcribe & Fix")  # New button
+            transcribe_fix_btn = gr.Button("Transcribe & Fix")  
+            save_transcribe_fix_btn = gr.Button("Save, Transcribe & Fix")  # New button
         
         with gr.Row():
             audio_output = gr.Audio(label="Saved Audio (MP3)", format="mp3")
@@ -340,6 +357,12 @@ with gr.Blocks(title="Audio Recorder, Transcriber, and Text Generator") as demo:
             inputs=[audio_input, prompt_type],
             outputs=[transcription_output, generated_text_output]
         )
+        
+        save_transcribe_fix_btn.click(
+            save_transcribe_fix,
+            inputs=[audio_input, prompt_type],
+            outputs=[audio_output, transcription_output, generated_text_output]
+        )  # New button action
     
         generate_btn.click(
             generate_text,
@@ -366,6 +389,6 @@ with gr.Blocks(title="Audio Recorder, Transcriber, and Text Generator") as demo:
     )
 
 # Launch the interface
-demo.launch() 
+demo.launch()