Remove chat history management, not needed

app.py

@@ -60,17 +60,16 @@ def transcribe_audio(audio_filepath):
 
 
 @spaces.GPU
-def transcript_qa(transcript, question, history, thinking_history):
+def transcript_qa(transcript, question):
     if not transcript:
-        return (history + [{"role": "user", "content": question}, {"role": "assistant", "content": "Please transcribe audio first before asking questions."}], 
-                thinking_history + "")
+        return "Please transcribe audio first before asking questions.", ""
     
     if not question or question.strip() == "":
-        return history, thinking_history
+        return "", ""
     
     with torch.inference_mode(), model.llm.disable_adapter():
         output_ids = model.generate(
-            prompts=[[{"role": "user", "content": f"Answer this question about the transcript: {question}\n\nTranscript: {transcript}"}]],
+            prompts=[[{"role": "user", "content": f"{question}\n\nTranscript: {transcript}"}]],
             max_new_tokens=512,
         )
     
@@ -88,12 +87,7 @@ def transcript_qa(transcript, question, history, thinking_history):
     if not ans:
         ans = "I couldn't generate a response. Please try rephrasing your question."
     
-    # Update thinking history if there's new thinking
-    if thinking:
-        thinking_history += f"\n\n**Question:** {question}\n**Thinking:** {thinking}"
-    
-    return (history + [{"role": "user", "content": question}, {"role": "assistant", "content": ans}],
-            thinking_history)
+    return ans, thinking
 
 def disable_transcribe():
     return gr.update(interactive=False)
@@ -128,7 +122,6 @@ with gr.Blocks(theme=theme) as demo:
 
     # State variables
     transcript_state = gr.State("")
-    thinking_state = gr.State("")
 
     # Example questions
     example_questions = ["Can you summarize this meeting?", "Please provide bullet points of the key items.",
@@ -179,11 +172,11 @@ with gr.Blocks(theme=theme) as demo:
             example_container = gr.Column()
         
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(
-                label="Chat History", 
-                type="messages", 
-                height=400,
-                show_copy_button=True
+            answer_box = gr.Textbox(
+                label="Assistant's Response",
+                lines=8,
+                max_lines=12,
+                interactive=False
             )
             with gr.Row():
                 msg = gr.Textbox(
@@ -214,19 +207,19 @@ with gr.Blocks(theme=theme) as demo:
         )
 
     # Event handlers
-    def submit_question(question, history, transcript, thinking):
+    def submit_question(question, transcript):
         if not question or question.strip() == "":
-            return "", history, thinking
-        new_history, new_thinking = transcript_qa(transcript, question, history, thinking)
-        return "", new_history, new_thinking
+            return "", "", ""
+        answer, thinking = transcript_qa(transcript, question)
+        return "", answer, thinking
 
 
     transcribe_btn.click(
         fn=disable_transcribe,
         outputs=[transcribe_btn]
     ).then(
-        fn=lambda: ([], "", ""),
-        outputs=[chatbot, thinking_box, thinking_state]
+        fn=lambda: ("", ""),
+        outputs=[answer_box, thinking_box]
     ).then(
         fn=transcribe_audio,
         inputs=[audio_input],
@@ -248,27 +241,19 @@ with gr.Blocks(theme=theme) as demo:
 
     msg.submit(
         fn=submit_question,
-        inputs=[msg, chatbot, transcript_state, thinking_state],
-        outputs=[msg, chatbot, thinking_state]
-    ).then(
-        fn=lambda x: x,
-        inputs=[thinking_state],
-        outputs=[thinking_box]
+        inputs=[msg, transcript_state],
+        outputs=[msg, answer_box, thinking_box]
     )
 
     submit_chat_btn.click(
         fn=submit_question,
-        inputs=[msg, chatbot, transcript_state, thinking_state],
-        outputs=[msg, chatbot, thinking_state]
-    ).then(
-        fn=lambda x: x,
-        inputs=[thinking_state],
-        outputs=[thinking_box]
+        inputs=[msg, transcript_state],
+        outputs=[msg, answer_box, thinking_box]
     )
 
     clear_chat_btn.click(
-        fn=lambda: ([], "", ""),
-        outputs=[chatbot, thinking_box, thinking_state]
+        fn=lambda: ("", ""),
+        outputs=[answer_box, thinking_box]
     )
 
 demo.queue()