Update app.py

app.py

@@ -13,6 +13,16 @@ from config import openai_api_key
 # Load the model
 llm = load_model(openai_api_key)
 
+def process_message(message, history, speaker_id):
+    interview = get_interview_instance()
+    response = interview.process_message(message)
+    return response
+
+def start_new_interview(general_impression, speaker_id):
+    interview = get_interview_instance()
+    interview.set_general_impression(general_impression)
+    return interview.start_interview()
+    
 def analyze_video(video_path, progress=gr.Progress()):
     start_time = time.time()
     if not video_path:
@@ -36,7 +46,8 @@ def analyze_video(video_path, progress=gr.Progress()):
     execution_time = end_time - start_time
 
     output_components = []  # transcript
-
+    chatbots = []
+    
     output_components.append(f"Completed in {int(execution_time)} seconds.")
     output_components.append(gr.Textbox(value=transcription, label="Transcript", lines=10, visible=True))
                 
@@ -44,6 +55,21 @@ def analyze_video(video_path, progress=gr.Progress()):
         print(speaker_id)
         speaker_explanations = explanations[speaker_id]
         speaker_general_impression = general_impressions[speaker_id]
+
+        with gr.TabItem(label=f'Interactive Interview'):
+            chatbot = gr.Chatbot(visible=True)
+            msg = gr.Textbox(visible=True)
+            clear = gr.Button("Clear", visible=True)
+            start_interview = gr.Button("Start Interview", visible=True)
+            chatbots.append(chatbot)
+
+            msg.submit(process_message, inputs=[msg, chatbot, gr.State(speaker_id)], outputs=[chatbot])
+            clear.click(lambda: None, None, chatbot, queue=False)
+            start_interview.click(
+                start_new_interview,
+                inputs=[gr.State(speaker_general_impression), gr.State(speaker_id)],
+                outputs=[chatbot]
+            )
         
         with gr.Tab(visible=True):
             with gr.TabItem(label=f'General Impression'):
@@ -73,15 +99,16 @@ def analyze_video(video_path, progress=gr.Progress()):
                     gr.Textbox(value=speaker_explanations.get("personality", ""),
                                label="Personality Disorders Explanation", visible=True, lines=2)
                 ]
-
+                
+        output_components.extend([chatbot, msg, clear, start_interview])
         output_components.extend(speaker_section1)
         output_components.extend(speaker_section2)
         output_components.extend(speaker_section3)
         output_components.extend(speaker_section4)
 
     # Pad with None for any missing speakers
-    while len(output_components) < 28:
-        output_components.extend([gr.update(visible=False)] * 9)
+    while len(output_components) < 40:  
+        output_components.extend([gr.update(visible=False)] * 13)
 
     return output_components
 
@@ -140,19 +167,20 @@ with gr.Blocks() as iface:
     for n in range(3):  # Assuming maximum of 3 speakers
 
         with gr.Tab(label=f'Speaker {n + 1}', visible=True):
+            
+            with gr.TabItem(label=f'Interactive Interview'):
+                column_components_interview = [
+                    gr.Chatbot(visible=False),
+                    gr.Textbox(visible=False),
+                    gr.Button("Clear", visible=False),
+                    gr.Button("Start Interview", visible=False)
+                ]
+                            
             with gr.TabItem(label=f'General Impression'):
                 column_components1 = [
                     gr.Markdown(visible=False),
                     gr.Textbox(label="General Impression")]
 
-            
-            with gr.Tab("Interactive Interview"):
-                chatbot = gr.Chatbot()
-                msg = gr.Textbox()
-                clear = gr.Button("Clear")
-                start_interview = gr.Button("Start Interview")
-            
-
             with gr.TabItem(label=f'Attachment Styles'):
                 column_components2 = [
                     gr.Plot(visible=False),
@@ -169,6 +197,7 @@ with gr.Blocks() as iface:
                     gr.Plot(visible=False),
                     gr.Textbox(label="Personality Disorders Explanation")]
 
+        output_components.extend(column_components_interview)
         output_components.extend(column_components1)
         output_components.extend(column_components2)
         output_components.extend(column_components3)