Retain chatbot history when chatting with video

.gitignore

@@ -210,4 +210,12 @@ $RECYCLE.BIN/
 *.msp
 
 # Windows shortcuts
-*.lnk
+*.lnk
+
+.vscode/*
+
+# Local History for Visual Studio Code
+.history/
+
+# Built Visual Studio Code Extensions
+*.vsix

app.py

@@ -2,7 +2,7 @@ import os
 import random
 from dataclasses import dataclass
 from time import sleep
-from typing import Dict, List, Generator, Optional
+from typing import Dict, List, Generator
 
 import cv2
 import gradio as gr
@@ -30,7 +30,7 @@ class MockInterviewer:
 
     def chat_with_text(
         self, 
-        message: Dict, 
+        message: str, 
         history: List[List],
         job_role: str, 
         company: str, 
@@ -46,6 +46,7 @@ class MockInterviewer:
     def chat_with_video(
         self, 
         video: str,
+        history: List[List],
         job_role: str, 
         company: str, 
         job_description: str,
@@ -53,7 +54,7 @@ class MockInterviewer:
         technical_count: int, 
         situational_count: int, 
         case_count: int
-    ) -> str:
+    ) -> List[List]:
         with open(video, 'rb') as file:
             transcriptions = self._client.audio.transcriptions.create(
                 model='whisper-1',
@@ -63,7 +64,8 @@ class MockInterviewer:
         os.remove(video)
         config = Config(job_role, company, job_description, behavioral_count, technical_count, situational_count, case_count)
         response = self._chat(transcriptions.text, config, video_frame_file_ids)
-        return [(transcriptions.text, response)]
+        history.append([transcriptions.text, response])
+        return history
 
     def clear_thread(self) -> None:
         print('Initializing new thread')
@@ -204,9 +206,9 @@ with gr.Blocks(theme=theme) as demo:
             chat_interface.clear_btn.click(mock_interviewer.clear_thread)
         
         with gr.Column(variant='panel', scale=1):
-            video = gr.Video(sources='webcam', include_audio=True)
+            video = gr.Video(sources=['webcam'], include_audio=True)
             video.stop_recording(fn=mock_interviewer.chat_with_video,
-                                inputs=[video, job_role, company, job_description, behavioral_count, technical_count, situational_count, case_count],
+                                inputs=[video, chat_interface.chatbot_state, job_role, company, job_description, behavioral_count, technical_count, situational_count, case_count],
                                 outputs=[chat_interface.chatbot],
                                 api_name=False).then(lambda: None, None, video, queue=False)