Cancel info from Gradio

app.py

@@ -8,46 +8,67 @@ import gradio as gr
 from google import genai
 from google.genai import types
 
+# ------------------------------------------------------------
+# Configuration
+# ------------------------------------------------------------
+
 # Supported input audio formats
 SUPPORTED_AUDIO_EXTS = {".wav", ".mp3", ".m4a"}
 
-# Default system prompt
+# Default models (can be overridden via environment variables if desired)
+DEFAULT_TEXT_MODEL = os.getenv("GEMINI_TEXT_MODEL", "gemini-2.0-flash")
+DEFAULT_TTS_MODEL = os.getenv("GEMINI_TTS_MODEL", "gemini-2.5-flash-preview-tts")
+
+# Default system prompt used to steer Gemini for meeting-notes behavior.
 DEFAULT_SYSTEM_PROMPT = (
     """Here are instructions from the user outlining your goals and how you should respond:
 You are a tool designed to assist users by enhancing the efficiency and effectiveness of their meeting notes output. You specialize in three primary functions:
 
+
 Structuring Notes
 Objective: Transform unorganized, rough draft notes into a structured, easy-to-navigate document.
 Process: Employ NLP and text analysis algorithms to identify key themes, decisions, action items, and insights from the rough notes. Then, organize these elements according to the predefined template that highlights the meeting's objectives, key points, decisions made, action items, and so on.
 
+
 Synthesizing Transcripts
 Objective: Convert lengthy, verbatim meeting transcripts into concise, coherent summaries.
 Process: Use advanced summarization techniques to extract the essence of discussions, focusing on outcomes, decisions, and action points. Integration of sentiment analysis could further highlight discussions' tone and areas of agreement or contention.
 
+
 Develop Downloadable Microsoft Word Documents
 Objective: Provide users with a convenient, editable format for the structured notes and synthesized summaries.
-Process: After structuring or synthesizing the content, format the output into a Microsoft Word document. 
+Process: After structuring or synthesizing the content, format the output into a Microsoft Word document.
+
 
 Here is a breakdown of your operational flow:
 
-Initiation: Each session with the Meeting Notes Assistant begins when a consultant initiates interaction by selecting the "Let's Begin" conversation starter prompt. 
 
-Upon activation, you will print the following text: 
+Initiation: Each session with the Meeting Notes Assistant begins when a consultant initiates interaction by selecting the "Let's Begin" conversation starter prompt.
+
+
+Upon activation, you will print the following text:
+
 
 "Welcome! Ready to streamline your meeting notes? Here's how I can help:
 
+
 1. Organize Draft Notes: For structured, coherent notes.
 2. Combine Notes: Merge notes from multiple sources into one document.
 3. Summarize Transcripts: Get concise summaries of your meetings.
 Just reply with 1, 2, or 3 to choose. Let's get started!"
 
 
+
+
 If the user selects 1 (Organize Draft Notes), then the tool responds with:
 
+
 "Great choice! Let's tidy up those draft notes. To get started, please paste your notes below or upload the file. I'll help structure them for clarity and coherence."
 
+
 Upon receiving the notes, you will structure the output with the following content. Do not use every piece of this, read the notes and make a decision on the best elements to add. Never use all of them:
 
+
 Meeting Title: [Automatically extracted or prompted for if unclear.]
 Date: [Automatically extracted or prompted for.]
 Attendees: [List of participants, extracted from notes or prompted for.]
@@ -59,22 +80,29 @@ Insights and Learnings: [Any notable insights gleaned from the meeting.]
 Next Steps: [Detailed actions and responsibilities, clearly outlined for follow-up.]
 Formatted Notes: [The structured output, presented in a professional, concise format on no more than one page, focusing on clarity, coherence, and actionable insights.]
 
+
 You will always print the following question at the end of your note synthesis: "Would you like for me to generate a downloadable Word document of your notes?"
 
-If the user says anything affirmative like "yes", then use Code Interpretor to generate a downloadable Word document of the content. 
+
+If the user says anything affirmative like "yes", then use Code Interpretor to generate a downloadable Word document of the content.
 If the user says anything like "no", then respond with this: "Great! Can I help you with any other notes? Here's how I can help:
 
+
 1. Organize Draft Notes: For structured, coherent notes.
 2. Combine Notes: Merge notes from multiple sources into one document.
 3. Summarize Transcripts: Get concise summaries of your meetings.
 Just reply with 1, 2, or 3 to choose!"
 
+
 If the user selects 2 (Combine and Synthesize Notes), then the tool responds with:
 
+
 "Perfect! Let's merge and synthesize your notes from multiple sources. Please paste the notes below or upload the files, and I'll create a unified, coherent document."
 
+
 After the user submits their notes, the tool processes and formats the output as follows:
 
+
 Meeting Title: [Consolidated from provided notes or prompted for if unclear.]
 Date(s): [Consolidated dates or prompted for if not provided.]
 Attendees: [Compiled list of all participants mentioned across sources.]
@@ -86,22 +114,29 @@ Insights and Learnings: [Notable insights drawn from the synthesis of all notes.
 Next Steps: [Consolidated list of detailed actions and responsibilities from all sources.]
 Recommendations: [Suggestions for next steps or considerations, based on the synthesized information.]
 
+
 You will always print the following question at the end of your note synthesis: "Would you like for me to generate a downloadable Word document of your notes?"
 
-If the user says anything affirmative like "yes", then use Code Interpretor to generate a downloadable Word document of the content. 
+
+If the user says anything affirmative like "yes", then use Code Interpretor to generate a downloadable Word document of the content.
 If the user says anything like "no", then respond with this: "Great! Can I help you with any other notes? Here's how I can help:
 
+
 1. Organize Draft Notes: For structured, coherent notes.
 2. Combine Notes: Merge notes from multiple sources into one document.
 3. Summarize Transcripts: Get concise summaries of your meetings.
 Just reply with 1, 2, or 3 to choose!"
 
+
 If the user selects 3 (Summarize a Meeting Transcript), then the tool responds with:
 
+
 "Excellent, let's summarize your meeting transcript. Please upload the transcript file or paste it into the chat window, and I'll distill it into a concise summary, capturing all critical points and actionable insights."
 
+
 The tool then processes the transcript and structures the output as follows:
 
+
 Meeting Title: [Title extracted or prompted for if unclear.]
 Date: [Date extracted or prompted for.]
 Attendees: [List of participants, extracted or prompted for clarification.]
@@ -116,22 +151,35 @@ Next Steps: [Detailed actions and responsibilities, outlined for clarity and acc
 Recommendations: [Strategic suggestions based on the meeting's content, aimed to add value and perspective for planning.]
 You will always print the following question at the end of your note synthesis: "Would you like for me to generate a downloadable Word document of your notes?"
 
-If the user says anything affirmative like "yes", then use Code Interpretor to generate a downloadable Word document of the content. 
+
+If the user says anything affirmative like "yes", then use Code Interpretor to generate a downloadable Word document of the content.
 If the user says anything like "no", then respond with this: "Great! Can I help you with any other notes? Here's how I can help:
 
+
 1. Organize Draft Notes: For structured, coherent notes.
 2. Combine Notes: Merge notes from multiple sources into one document.
 3. Summarize Transcripts: Get concise summaries of your meetings.
 Just reply with 1, 2, or 3 to choose!"
 
+
 You will maintain consistency and apply this approach every time. This approach ensures notes are concise, and actionable, maintaining a professional tone throughout. Your recommendations are provided to enhance strategic outcomes and decision-making processes, always tailored to the context of the discussion."""
 )
 
+# ------------------------------------------------------------
+# Utility functions
+# ------------------------------------------------------------
 
-def wave_file(filename: str, pcm: bytes, channels: int = 1,
-              rate: int = 24000, sample_width: int = 2) -> None:
+
+def wave_file(
+    filename: str,
+    pcm: bytes,
+    channels: int = 1,
+    rate: int = 24000,
+    sample_width: int = 2,
+) -> None:
     """
     Save raw PCM data as a WAV file.
+    Mirrors the helper used in the official Gemini TTS examples.
     """
     with wave.open(filename, "wb") as wf:
         wf.setnchannels(channels)
@@ -187,51 +235,77 @@ def normalize_audio_file(audio_path: str) -> str:
 
 
 def generate_text_with_optional_audio(
+    client: genai.Client,
     model: str,
     prompt: str,
     system_prompt: Optional[str] = None,
     audio_path: Optional[str] = None,
 ) -> str:
     """
-    Call Gemini with text prompt and optional audio file.
+    Call Gemini with a text prompt and an optional audio file.
+    Uses the modern google-genai Client and models.generate_content API.
     Returns the model's text response.
     """
-    contents = [prompt]
+    contents: list = [prompt]
 
     if audio_path:
         normalized = normalize_audio_file(audio_path)
-        uploaded_file = genai.upload_file(normalized)
+        uploaded_file = client.files.upload(file=normalized)
         contents.append(uploaded_file)
 
     effective_system_prompt = system_prompt or DEFAULT_SYSTEM_PROMPT
 
-    model_instance = genai.GenerativeModel(
-        model_name=model,
-        system_instruction=effective_system_prompt,
+    response = client.models.generate_content(
+        model=model,
+        contents=contents,
+        config=types.GenerateContentConfig(
+            system_instruction=effective_system_prompt,
+        ),
     )
-    
-    response = model_instance.generate_content(contents)
-    return response.text
+
+    return response.text or ""
 
 
 def tts_from_text(
+    client: genai.Client,
     text: str,
-    model: str = "gemini-1.5-flash",
+    model: str = DEFAULT_TTS_MODEL,
     voice_name: str = "Kore",
 ) -> str:
     """
-    Convert text to speech using Gemini TTS and save to a WAV file.
-    Returns the path to the generated audio file.
-    Note: TTS is not yet available in google.generativeai, returning placeholder.
+    Convert text to speech using Gemini native TTS (2.5 Flash TTS).
+    Uses the official speech-generation pattern with response_modalities=['AUDIO']
+    and SpeechConfig / VoiceConfig.
+
+    Returns the path to a temporary WAV file that Gradio can play.
     """
-    # TTS functionality is not yet available in the standard google.generativeai package
-    # This is a placeholder that returns None
-    # Users should use alternative TTS services like Google Cloud TTS or gTTS
-    raise NotImplementedError(
-        "TTS is not yet available in google.generativeai. "
-        "Please disable TTS output or use alternative TTS services."
+    response = client.models.generate_content(
+        model=model,
+        contents=text,
+        config=types.GenerateContentConfig(
+            response_modalities=["AUDIO"],
+            speech_config=types.SpeechConfig(
+                voice_config=types.VoiceConfig(
+                    prebuilt_voice_config=types.PrebuiltVoiceConfig(
+                        voice_name=voice_name
+                    )
+                )
+            ),
+        ),
     )
 
+    # Inline audio data is returned as raw PCM bytes that we wrap into a .wav file.
+    pcm_data = response.candidates[0].content.parts[0].inline_data.data
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
+        wave_file(tmp.name, pcm_data)
+        return tmp.name
+
+
+# ------------------------------------------------------------
+# Core meeting-notes processing
+# ------------------------------------------------------------
+
 
 def process_meeting_notes(
     operation_type: str,
@@ -239,187 +313,259 @@ def process_meeting_notes(
     audio_input: Optional[str],
     enable_tts: bool,
     api_key: str,
-    progress=gr.Progress()
+    progress=gr.Progress(),
 ) -> Tuple[str, Optional[str]]:
     """
     Main processing function for meeting notes.
-    Returns (text_output, audio_output_path)
+    Returns (text_output, audio_output_path).
     """
-    if not api_key:
-        return "❌ Error: Please provide your Gemini API key.", None
-    
+    # Determine effective API key: prefer user entry, fall back to env var.
+    effective_api_key = (api_key or "").strip() or os.getenv("GEMINI_API_KEY")
+
+    if not effective_api_key:
+        return (
+            "❌ Error: Please provide your Gemini API key (textbox) "
+            "or set GEMINI_API_KEY in the environment.",
+            None,
+        )
+
     if not text_input and not audio_input:
         return "❌ Error: Please provide either text input or an audio file.", None
-    
+
     try:
         # Initialize Gemini client
         progress(0.1, desc="Initializing Gemini client...")
-        genai.configure(api_key=api_key)
-        
+        client = genai.Client(api_key=effective_api_key)
+
         # Prepare prompt based on operation type
         progress(0.2, desc="Preparing prompt...")
         if operation_type == "Organize Draft Notes":
-            prompt = f"Please organize these draft notes:\n\n{text_input}" if text_input else "Please organize the notes from the audio file."
+            if text_input:
+                prompt = f"Please organize these draft notes:\n\n{text_input}"
+            else:
+                prompt = (
+                    "Please listen carefully to this meeting recording and generate "
+                    "organized draft notes following the system instructions."
+                )
         elif operation_type == "Combine Notes":
-            prompt = f"Please combine and synthesize these notes:\n\n{text_input}" if text_input else "Please combine and synthesize the notes from the audio file."
-        else:  # Summarize Transcripts
-            prompt = f"Please summarize this meeting transcript:\n\n{text_input}" if text_input else "Please summarize the meeting transcript from the audio file."
-        
+            if text_input:
+                prompt = (
+                    "Please combine and synthesize the following multi-source notes "
+                    "into one unified, coherent meeting document:\n\n"
+                    f"{text_input}"
+                )
+            else:
+                prompt = (
+                    "Please combine and synthesize the notes implied by this audio "
+                    "input into one coherent document following the system instructions."
+                )
+        else:  # "Summarize Transcripts"
+            if text_input:
+                prompt = (
+                    "Please summarize the following meeting transcript into concise, "
+                    "actionable notes:\n\n" f"{text_input}"
+                )
+            else:
+                prompt = (
+                    "Please summarize the meeting transcript contained in this audio "
+                    "file into concise, actionable notes."
+                )
+
         # Generate text response
         progress(0.4, desc="Processing with Gemini...")
         text_response = generate_text_with_optional_audio(
-            model="gemini-1.5-flash",
+            client=client,
+            model=DEFAULT_TEXT_MODEL,
             prompt=prompt,
             system_prompt=DEFAULT_SYSTEM_PROMPT,
             audio_path=audio_input,
         )
-        
+
         progress(0.7, desc="Text processing complete!")
-        
+
         # Generate TTS if enabled
-        audio_output = None
+        audio_output: Optional[str] = None
         if enable_tts:
-            progress(0.8, desc="Generating audio output...")
+            progress(0.8, desc="Generating audio output (TTS)...")
             try:
                 audio_output = tts_from_text(
+                    client=client,
                     text=text_response,
-                    model="gemini-1.5-flash",
+                    model=DEFAULT_TTS_MODEL,
                     voice_name="Kore",
                 )
                 progress(1.0, desc="Audio generation complete!")
-            except NotImplementedError as tts_error:
-                progress(1.0, desc="Processing complete (TTS unavailable)")
-                return text_response + "\n\n⚠️ Note: " + str(tts_error), None
+            except Exception as tts_error:
+                progress(1.0, desc="Processing complete (TTS error)")
+                return (
+                    text_response
+                    + "\n\n⚠️ Note: TTS generation failed with:\n"
+                    + str(tts_error),
+                    None,
+                )
         else:
             progress(1.0, desc="Processing complete!")
-        
+
         return text_response, audio_output
-    
+
     except Exception as e:
-        return f"❌ Error: {str(e)}", None
+        return f"❌ Error while processing: {str(e)}", None
+
 
+# ------------------------------------------------------------
+# Gradio UI
+# ------------------------------------------------------------
 
-# Create Gradio interface
 with gr.Blocks(title="Meeting Notes Assistant", theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # 📝 Meeting Notes Assistant
-        
+
         Transform your meeting notes with AI-powered organization, synthesis, and summarization.
-        
+
         **Features:**
         - 📋 Organize Draft Notes
         - 🔄 Combine Multiple Notes
         - 📊 Summarize Transcripts
         - 🎤 Audio Input Support (WAV, MP3, M4A)
-        - 🔊 Text-to-Speech Output
+        - 🔊 Text-to-Speech Output (Gemini 2.5 TTS)
         """
     )
-    
+
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### ⚙️ Configuration")
-            
+
             api_key_input = gr.Textbox(
-                label="Gemini API Key",
-                placeholder="Enter your Gemini API key",
+                label="Gemini API Key (optional)",
+                placeholder="Enter your Gemini API key, or leave blank to use GEMINI_API_KEY env var",
                 type="password",
-                info="Get your API key from Google AI Studio"
             )
-            
+            gr.Markdown(
+                "🔑 Get your API key from Google AI Studio and optionally store it "
+                "as the GEMINI_API_KEY environment variable on this Space."
+            )
+
             operation_type = gr.Radio(
-                choices=["Organize Draft Notes", "Combine Notes", "Summarize Transcripts"],
+                choices=[
+                    "Organize Draft Notes",
+                    "Combine Notes",
+                    "Summarize Transcripts",
+                ],
                 label="Operation Type",
                 value="Organize Draft Notes",
-                info="Select the type of operation you want to perform"
             )
-            
+            gr.Markdown(
+                "ℹ️ Choose how you want the assistant to process your input: "
+                "organize messy notes, merge multiple notes, or summarize transcripts."
+            )
+
             enable_tts = gr.Checkbox(
                 label="Enable Text-to-Speech Output",
                 value=False,
-                info="Generate audio output of the processed notes"
             )
-        
+            gr.Markdown(
+                "🎧 When enabled, the assistant will also generate a spoken version "
+                "of the processed notes using Gemini TTS."
+            )
+
         with gr.Column(scale=2):
             gr.Markdown("### 📥 Input")
-            
+
             text_input = gr.Textbox(
                 label="Text Input",
                 placeholder="Paste your meeting notes, transcript, or text here...",
                 lines=10,
-                info="Enter your text or leave empty if using audio input"
             )
-            
+            gr.Markdown(
+                "✍️ Enter any text-based notes or transcripts here. "
+                "You can also leave this empty and only upload audio."
+            )
+
             audio_input = gr.Audio(
                 label="Audio Input (Optional)",
                 type="filepath",
-                info="Upload WAV, MP3, or M4A file"
             )
-            
-            process_btn = gr.Button("🚀 Process Notes", variant="primary", size="lg")
-    
+            gr.Markdown(
+                "🎙️ Upload a WAV, MP3, or M4A file. M4A will be converted "
+                "to WAV automatically (requires ffmpeg in the environment)."
+            )
+
+            process_btn = gr.Button("🚀 Process Notes", variant="primary")
+
     gr.Markdown("### 📤 Output")
-    
+
     with gr.Row():
         with gr.Column():
             text_output = gr.Textbox(
                 label="Processed Notes",
                 lines=15,
                 show_copy_button=True,
-                info="Your organized, combined, or summarized notes will appear here"
             )
-        
+            gr.Markdown(
+                "✅ Your organized, combined, or summarized notes will appear here. "
+                "You can copy them with the button in the top-right corner."
+            )
+
         with gr.Column():
             audio_output = gr.Audio(
                 label="Audio Output",
                 type="filepath",
-                info="Generated audio (if TTS is enabled)"
             )
-    
+            gr.Markdown(
+                "🔊 If Text-to-Speech is enabled and succeeds, the generated audio "
+                "will appear here for playback or download."
+            )
+
     # Example inputs
     gr.Markdown("### 💡 Examples")
     gr.Examples(
         examples=[
             [
                 "Organize Draft Notes",
-                "Meeting with John about Q4 planning. Discussed budget allocation. Need to finalize numbers by Friday. Action items: Sarah to review projections, Mike to update spreadsheet.",
+                "Meeting with John about Q4 planning. Discussed budget allocation. "
+                "Need to finalize numbers by Friday. Action items: Sarah to review "
+                "projections, Mike to update spreadsheet.",
                 None,
-                False
+                False,
             ],
             [
                 "Summarize Transcripts",
-                "John: We need to discuss the Q4 budget. Sarah: I think we should allocate more to marketing. Mike: I agree, but we need to be careful with spending. John: Let's review the numbers and decide by Friday.",
+                "John: We need to discuss the Q4 budget. Sarah: I think we should "
+                "allocate more to marketing. Mike: I agree, but we need to be careful "
+                "with spending. John: Let's review the numbers and decide by Friday.",
                 None,
-                False
+                False,
             ],
         ],
         inputs=[operation_type, text_input, audio_input, enable_tts],
     )
-    
+
     # Connect the process button
     process_btn.click(
         fn=process_meeting_notes,
         inputs=[operation_type, text_input, audio_input, enable_tts, api_key_input],
         outputs=[text_output, audio_output],
     )
-    
+
     gr.Markdown(
         """
         ---
         ### 📚 How to Use
-        
-        1. **Enter your Gemini API Key** (required)
-        2. **Select operation type**: Organize, Combine, or Summarize
-        3. **Provide input**: Enter text or upload audio (or both)
-        4. **Optional**: Enable TTS for audio output
-        5. **Click Process Notes** and wait for results
-        
+
+        1. **Enter your Gemini API Key** in the box, or configure `GEMINI_API_KEY` in the environment.
+        2. **Select operation type**: Organize, Combine, or Summarize.
+        3. **Provide input**: Enter text and/or upload audio.
+        4. **Optional**: Enable TTS for audio output.
+        5. **Click “Process Notes”** and wait for results.
+
         ### 🔑 Getting Your API Key
-        
-        Visit [Google AI Studio](https://aistudio.google.com/app/apikey) to get your free Gemini API key.
+
+        Visit Google AI Studio to get your free Gemini API key.
         """
     )
 
 
 if __name__ == "__main__":
-    demo.launch()
+    # Enable queuing so that gr.Progress works correctly in Spaces and other deployments.
+    demo.queue().launch()