Upload 4 files

app.py

@@ -21,6 +21,31 @@ temp_audio_files = []
 initial_audio_path = None
 
 
+def reset_interview_action():
+    global question_count, interview_history
+    question_count = 0
+    interview_history.clear()
+    initial_message = get_interview_initial_message()
+
+    # Generate new audio for the initial message
+    initial_audio_buffer = BytesIO()
+    convert_text_to_speech(initial_message, initial_audio_buffer)
+    initial_audio_buffer.seek(0)
+
+    with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as temp_file:
+        temp_audio_path = temp_file.name
+        temp_file.write(initial_audio_buffer.getvalue())
+
+    temp_audio_files.append(temp_audio_path)
+
+    return (
+        [(None, initial_message)],  # Reset chatbot
+        temp_audio_path,  # New audio
+        gr.File(visible=False),  # Hide PDF output
+        gr.Textbox(visible=True),  # Show message input
+        "Interview reset. You can start a new interview now."  # Status message
+    )
+
 # Initialize Gradio interface
 def create_app():
     global initial_audio_path
@@ -55,10 +80,11 @@ def create_app():
                 visible=False,
                 show_download_button=False,
             )
+
+            reset_button = gr.Button("Reset Interview", size='sm')
             chatbot = gr.Chatbot(value=[(None, f"{initial_message}")], label=f"Clinical Interview 𝚿📋")
             msg = gr.Textbox(label="Type your message here...")
             send_button = gr.Button("Send")
-
             pdf_output = gr.File(label="Download Report", visible=False)
 
             def user(user_message, history):
@@ -118,6 +144,12 @@ def create_app():
                 bot_response, [chatbot, msg], [chatbot, audio_output, pdf_output, msg]
             )
 
+            reset_button.click(
+                reset_interview_action,
+                inputs=[],
+                outputs=[chatbot, audio_output, pdf_output, msg]
+            )
+
         with gr.Tab("Upload Document"):
             file_input = gr.File(label="Upload a TXT, PDF, or DOCX file")
             language_input = gr.Textbox(label="Preferred Language for Report",

prompt_instructions.py

@@ -11,12 +11,11 @@ def get_interview_initial_message():
     
     Feel free to share as much or as little as you're comfortable with. There are no right or wrong answers - I'm here to listen and learn about your experiences.
     
-    To ensure I can communicate with you most effectively, could you please tell me which language you prefer to speak or conduct this interview in?"""
+    Could you please tell me which language you prefer to speak or conduct this interview in?"""
 
 def get_interview_prompt(language, n_of_questions):
     return f"""You are a Psychologist or Psychiatrist conducting a clinical interview in {language}. 
 Use the following context and interview history to guide your response. 
-Keep your responses concise and to the point:
 
 Context from knowledge base: {{context}}
 
@@ -26,11 +25,9 @@ Previous interview history:
 Current question number: {{question_number}}
 
 Respond to the patient's input briefly and directly in {language}.
-Ask a specific, detailed question that hasn't been asked before, and keep it short.
-Do not repeat the same questions.
+Ask a specific, detailed question that hasn't been asked before.
 When asking questions, the way the questions are asked must take into account the patient's personality.
 For example, if the person is more introverted or extraverted, the way the questions are asked will be accordingly.
-If the person is very sensitive for example, there is a need to take that into consideration when asking the questions.
 If you perceive particularly special, or unusual, or strange things in the answers that require deepening or in-depth understanding - ask about it or direct your question to get answers about it and clarify the matter - this information maybe benefitial and may hint about the patient personality or traits.
 The first few questions are general questions about the patient that can give us an overall view.
 The 1st question is to ask for name.