Update app.py

app.py

@@ -136,6 +136,9 @@ def clear_chat_and_transcript(client_id):
         connections[client_id].transcript = ""
     return [], "", None, None
 
+def toggle_record_visibility(is_visible):
+    return not is_visible, gr.update(visible=not is_visible)
+
 # UI
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("# 📄 Document AI Assistant")
@@ -147,7 +150,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         padding: 10px 24px !important;
         margin-top: 6px;
     }
-    .record-button button {
+    #audio-stream button {
         display: none !important;
     }
     </style>
@@ -170,26 +173,36 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
                 send_btn = gr.Button("Send", variant="primary", scale=2)
 
             with gr.Accordion("🎤 Voice Transcription", open=False) as voice_section:
-                voice_input = gr.Audio(label="🎙️ Record", streaming=True, elem_classes="record-button")
+                voice_input = gr.Audio(label="🎙️ Mic Input", streaming=True, visible=False, elem_id="audio-stream")
                 voice_transcript = gr.Textbox(label="Transcript", lines=2, interactive=False)
 
                 with gr.Row():
+                    record_toggle_btn = gr.Button("🎙️ Record", elem_id="record-btn")
                     ask_btn = gr.Button("🟢 Ask", elem_id="ask-btn")
                     clear_chat_btn = gr.Button("🧹 Clear Chat", elem_id="clear-chat-btn")
-                    record_btn = gr.Button("🎙️ Record", elem_id="record-btn")
 
     # Functional bindings
     send_btn.click(fn=handle_chat,
                    inputs=[user_prompt, chat_state, thread_state, image_state],
                    outputs=[user_prompt, chat, thread_state, image_state])
     image_state.change(fn=lambda x: x, inputs=image_state, outputs=image_display)
+
     voice_input.stream(fn=send_audio, inputs=[voice_input, client_id], outputs=voice_transcript, stream_every=0.5)
+
+    record_toggle_btn.click(
+        fn=toggle_record_visibility,
+        inputs=[voice_enabled],
+        outputs=[voice_enabled, voice_input]
+    )
+
     ask_btn.click(fn=send_transcript_to_assistant,
                   inputs=[voice_transcript, chat_state, thread_state, image_state],
                   outputs=[user_prompt, chat, thread_state, image_state])
+
     clear_chat_btn.click(fn=clear_chat_and_transcript,
                          inputs=[client_id],
                          outputs=[chat, voice_transcript, thread_state, image_state])
+
     app.load(fn=create_ws, outputs=[client_id])
 
 app.launch()