Modiy UI and properly handle chat state

app.py

@@ -64,7 +64,7 @@ def transcript_qa(transcript, question, history):
     if not transcript:
         return history + [{"role": "user", "content": question}, {"role": "assistant", "content": "Please transcribe audio first before asking questions."}]
     
-    if not question:
+    if not question or question.strip() == "":
         return history
     
     with torch.inference_mode(), model.llm.disable_adapter():
@@ -82,6 +82,9 @@ def transcript_qa(transcript, question, history):
             _, ans = ans.split("</think>")
     ans = ans.strip()
     
+    if not ans:
+        ans = "I couldn't generate a response. Please try rephrasing your question."
+    
     return history + [{"role": "user", "content": question}, {"role": "assistant", "content": ans}]
 
 def disable_transcribe():
@@ -150,34 +153,49 @@ with gr.Blocks(theme=theme) as demo:
             clear_transcript_btn = gr.Button("Clear Transcript")
 
     # Spacing
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown()
-            gr.Markdown("---")
-            gr.Markdown()
+    gr.Markdown("---")
 
     # Step 3 - Interactive Q&A
-    with gr.Row(equal_height=True):
+    gr.Markdown("### Step 3 - Interactive Q&A")
+    
+    with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### Step 3 - Interactive Q&A")
-            msg = gr.Textbox(
-                placeholder="Ask a question about the transcript or pick an example", 
-                label="Questions")
-            gr.Examples(
-                examples=example_questions,
-                inputs=msg
+            gr.Markdown("#### About Context-Aware Q&A")
+            gr.Markdown("""The model retains the full transcript context, allowing you to ask follow-up questions 
+                        naturally without re-stating information. It understands references like 'they', 'it', or 'that topic'.""")
+            gr.Markdown("#### Example Questions")
+        
+        with gr.Column(scale=2):
+            chatbot = gr.Chatbot(
+                label="Chat History", 
+                type="messages", 
+                height=400,
+                show_copy_button=True
             )
-            submit_chat_btn = gr.Button("Submit", variant="primary")
-            clear_chat_btn = gr.Button("Clear Chat")
+            with gr.Row():
+                msg = gr.Textbox(
+                    placeholder="Ask a question about the transcript...", 
+                    label="Your Question",
+                    scale=4
+                )
+                submit_chat_btn = gr.Button("Send", variant="primary", scale=1)
+            clear_chat_btn = gr.Button("Clear Chat", size="sm")
     
+    # Add examples after msg is defined
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### Answers")
-            chatbot = gr.Chatbot(label="Responses", type="messages", height=400)
+            gr.Examples(
+                examples=example_questions,
+                inputs=msg,
+                cache_examples=False,
+                label=""
+            )
+        with gr.Column(scale=2):
+            pass  # Empty column for alignment
 
     # Event handlers
     def submit_question(question, history, transcript):
-        if not question:
+        if not question or question.strip() == "":
             return "", history
         new_history = transcript_qa(transcript, question, history)
         return "", new_history