Chore: Remove Chatbot, add chatInterface, Remove buttons that are not needed with new interface and handlers

app.py

@@ -108,11 +108,11 @@ def transcript_qa(transcript, question, history):
             import time
             time.sleep(0.01)
 
-def disable_buttons():
-    return gr.update(interactive=False), gr.update(interactive=False)
+def disable_transcribe():
+    return gr.update(interactive=False)
 
-def enable_buttons():
-    return gr.update(interactive=True), gr.update(interactive=True)
+def enable_transcribe():
+    return gr.update(interactive=True)
 
 # Build the Gradio interface
 with gr.Blocks(theme=theme) as demo:
@@ -200,60 +200,71 @@ with gr.Blocks(theme=theme) as demo:
             )
 
     gr.Markdown("### Step3 - Interactive Q&A")
-    chatbot = gr.ChatInterface("Q&A",
+    
+    # Create a wrapper function for ChatInterface
+    def qa_wrapper(message, history):
+        # Get the transcript from state
+        if transcript_state.value:
+            # Convert history format if needed
+            formatted_history = []
+            for msg in history:
+                if isinstance(msg, dict):
+                    formatted_history.append(msg)
+                else:
+                    # Handle tuple format if needed
+                    formatted_history.append({"role": "user", "content": msg[0]})
+                    if len(msg) > 1 and msg[1]:
+                        formatted_history.append({"role": "assistant", "content": msg[1]})
+            
+            # Process the Q&A
+            for response_history, _ in transcript_qa(transcript_state.value, message, formatted_history):
+                # Return just the last assistant message
+                if response_history and response_history[-1]["role"] == "assistant":
+                    yield response_history[-1]["content"]
+        else:
+            yield "Please transcribe audio first before asking questions."
+    
+    # Use ChatInterface for cleaner UI
+    chatbot = gr.ChatInterface(
+        fn=qa_wrapper,
         type="messages",
-        chatbot=gr.Chatbot(height=450),
-        textbox=gr.Textbox(placeholder="Type your message here..."),
+        chatbot=gr.Chatbot(
+            height=450,
+            label="",
+            bubble_full_width=False
+        ),
+        textbox=gr.Textbox(
+            placeholder="Ask a question about the transcript...",
+            container=False,
+            scale=7
+        ),
         examples=[
             "Can you please summarize this?",
             "What were the key points discussed?",
             "What was the main topic?",
             "What is the TLDR version so I can just leave this conference call early?"
         ],
-        run_examples_on_click=True,
-        cache_examples=True,
-        cache_mode="lazy"
+        submit_btn="Ask",
+        retry_btn=None,
+        undo_btn=None,
+        clear_btn="Clear"
     )
-    
-    def user(user_message, history: list):
-        return "", history + [{"role": "user", "content": user_message}]
-    
-    with gr.Row():
-        ask_btn = gr.Button("Ask", variant="primary", scale=1, size="lg")
-        clear_chat_btn = gr.Button("Clear", variant="secondary", scale=1, size="lg")
 
-    # Event handlers
+    # Event handlers - simplified since ChatInterface handles Q&A
     transcribe_btn.click(
-        fn=disable_buttons,
+        fn=disable_transcribe,
         inputs=None,
-        outputs=[transcribe_btn, ask_btn]
+        outputs=[transcribe_btn]
     ).then(
         fn=transcribe_audio,
         inputs=[audio_input],
-        outputs=[transcript_output, transcript_state, chatbot]
+        outputs=[transcript_output, transcript_state, chatbot.chatbot]  # Update the chatbot component
     ).then(
-        fn=enable_buttons,
+        fn=enable_transcribe,
         inputs=None,
-        outputs=[transcribe_btn, ask_btn]
-    )
-    
-    ask_btn.click(
-        fn=transcript_qa,
-        inputs=[transcript_state, question_input, chatbot],
-        outputs=[chatbot, question_input]
-    )
-    
-    question_input.submit(
-        fn=transcript_qa,
-        inputs=[transcript_state, question_input, chatbot],
-        outputs=[chatbot, question_input]
-    )
-    
-    clear_chat_btn.click(
-        fn=lambda: [],
-        inputs=None,
-        outputs=chatbot
+        outputs=[transcribe_btn]
     )
 
 demo.queue()
-demo.launch()
+demo.launch()
+