Update pages/02_Take_Interview.py

pages/02_Take_Interview.py

@@ -1,65 +1,25 @@
 import streamlit as st
-import sounddevice as sd
-import soundfile as sf
-import numpy as np
-import scipy.io.wavfile as wav
 import openai
 from langchain_openai import ChatOpenAI
 import os
 import io
-import time
-import soundfile as sf
-from openai import OpenAI
-
-client = OpenAI()
+import datetime
+from audio_recorder_streamlit import audio_recorder
 
 # Set OpenAI API key
-os.environ["OPENAI_API_KEY"] = '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'
+os.environ["OPENAI_API_KEY"] = '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'
+client = openai.OpenAI()
 
-# Session state variables to control recording status
-if "recording" not in st.session_state:
-    st.session_state.recording = False
-if "audio_data" not in st.session_state:
-    st.session_state.audio_data = None
+# Initialize session state variables
 if "responses" not in st.session_state:
     st.session_state.responses = []
+if "questions" not in st.session_state:
+    st.session_state.questions = []
+if "insights" not in st.session_state:
+    st.session_state.insights = "Candidate insights text here."
+if "jd_text" not in st.session_state:
+    st.session_state.jd_text = "Job description text here."
 
-# Function to record audio using sounddevice
-def start_recording(file_path, duration=10, fs=44100, device_index=0):
-    st.session_state.recording = True
-    st.write(f"Recording Started....Duration is {duration} secs.")
-    audio_data = sd.rec(int(duration * fs), samplerate=fs, channels=1, dtype='int16', device=device_index)
-    sd.wait()  # Wait until the recording is finished
-    st.session_state.recording = False
-    sf.write(file_path, audio_data, fs)
-    st.write(f"Recording complete. File saved at {file_path}")
-
-    # Convert to BytesIO object in WAV format
-    audio_buffer = io.BytesIO()
-    wav.write(audio_buffer, fs, audio_data)
-    audio_buffer.seek(0)  # Ensure the buffer starts at the beginning
-    st.session_state.audio_data = audio_buffer
-
-# Function to transcribe audio using OpenAI's Whisper API
-# Function to transcribe audio using OpenAI's Whisper API
-def transcribe_audio(file_path):
-    with open(file_path, 'rb') as audio_file:
-        transcription = client.audio.transcriptions.create(
-            model="whisper-1",
-            file=audio_file,
-            response_format="text"
-        )
-
-    # Print transcription to verify the format
-    st.write("Transcription response:", transcription)
-
-    # Check if transcription is a string or dictionary
-    if isinstance(transcription, dict) and 'text' in transcription:
-        return transcription['text']
-    elif isinstance(transcription, str):
-        return transcription  # Return the string if it's already in text format
-    else:
-        return "Error: Unexpected response format."
 
 # Function to generate questions based on insights and job description
 def generate_questions(insights, jd_text):
@@ -79,6 +39,27 @@ def generate_questions(insights, jd_text):
     questions = response.generations[0][0].text.strip().split('\n')
     return questions
 
+
+# Function to save the audio file
+def save_audio_file(audio_bytes, file_extension="wav"):
+    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+    file_name = f"audio_{timestamp}.{file_extension}"
+    file_path = os.path.join("saved_audios", file_name)
+
+    os.makedirs("saved_audios", exist_ok=True)  # Ensure the directory exists
+
+    with open(file_path, "wb") as f:
+        f.write(audio_bytes)
+    return file_path
+
+
+# Function to transcribe audio using Whisper API
+def transcribe_audio(audio_bytes):
+    audio_file = io.BytesIO(audio_bytes)
+    response = openai.Audio.transcribe("whisper-1", audio_file)
+    return response
+
+
 # Function to analyze the transcribed response
 def analyze_response(transcript):
     prompt = f"""
@@ -96,59 +77,68 @@ def analyze_response(transcript):
     analysis = response.generations[0][0].text.strip()
     return analysis
 
+
+# Function to handle audio recording and transcription
+def handle_audio_recording(question_index):
+    audio_bytes = audio_recorder(pause_threshold=2.0, sample_rate=41000, key=f"audio_recorder_{question_index}")
+    if audio_bytes:
+        st.audio(audio_bytes, format="audio/wav")  # Play recorded audio
+
+        # Save audio and provide feedback on file path
+        file_path = save_audio_file(audio_bytes, "wav")
+        st.write(f"Audio saved to: {file_path}")
+
+        # Transcribe the audio and analyze the response
+        transcript = transcribe_audio(audio_bytes)  # Transcribe using Whisper
+        st.write("### Transcription of Your Response")
+        st.write(transcript)
+
+        analysis = analyze_response(transcript)
+        st.write("### Analysis of Your Response")
+        st.write(analysis)
+
+        return transcript, analysis
+    return None, None
+
+
 st.title("Interview Preparation")
 
 # Display candidate insights and job description
-if "insights" in st.session_state and "jd_text" in st.session_state:
-    st.write("### Candidate Insights")
-    st.write(st.session_state.insights)
-
-    st.write("### Job Description")
-    st.write(st.session_state.jd_text)
-
-    # Generate interview questions
-    if "questions" not in st.session_state:
-        st.session_state.questions = generate_questions(st.session_state.insights, st.session_state.jd_text)
-
-    st.write("### Interview Questions")
-    responses = []
-    for i, question in enumerate(st.session_state.questions, 1):
-        st.write(f"**Question {i}:** {question}")
-
-        # Record or input text response
-        col1, col2 = st.columns(2)
-        with col1:
-            file_path = f"recording_question_{i}.wav"
-            # Start and Stop buttons for recording
-            if st.button(f"Start Answer for Question {i}"):
-                st.write(f"Starting to record for Question {i}...")
-                start_recording(file_path=file_path, duration=10)
-
-            if st.button(f"Analyze Answer for Question {i}") and st.session_state.audio_data:
-                transcript = transcribe_audio(file_path)
-
-                # Display the transcribed response
-                st.write("### Transcription of Your Response")
-                st.write(transcript)
-
-                # Analyze the transcribed response
-                analysis = analyze_response(transcript)
-                st.write("### Analysis of Your Response")
-                st.write(analysis)
-                responses.append((transcript, analysis))
-
-        with col2:
-            text_response = st.text_area(f"Or type your answer for Question {i}")
-            if text_response:
-                st.write("### Analysis of Your Response")
-                analysis = analyze_response(text_response)
-                st.write(analysis)
-                responses.append((text_response, analysis))
-
-    # Save responses for interviewer insights
-    st.session_state.responses = responses
-
-# Submit interview and generate final interviewer insights
+st.write("### Candidate Insights")
+st.write(st.session_state.insights)
+
+st.write("### Job Description")
+st.write(st.session_state.jd_text)
+
+# Generate interview questions if they haven't been generated yet
+if not st.session_state.questions:
+    st.session_state.questions = generate_questions(st.session_state.insights, st.session_state.jd_text)
+
+# Display and interact with each question
+responses = []
+for i, question in enumerate(st.session_state.questions, 1):
+    st.write(f"**Question {i}:** {question}")
+
+    col1, col2 = st.columns(2)
+    with col1:
+        st.write("#### Record Your Answer")
+        transcript, analysis = handle_audio_recording(i)  # Pass the question index to avoid duplicate IDs
+        if transcript and analysis:
+            responses.append((transcript, analysis))
+
+    with col2:
+        st.write("#### Or Type Your Answer")
+        text_response = st.text_area(f"Type your answer for Question {i}", key=f"text_response_{i}")
+        if text_response:
+            analysis = analyze_response(text_response)
+            st.write("### Analysis of Your Response")
+            st.write(analysis)
+            responses.append((text_response, analysis))
+
+# Save responses in session state
+st.session_state.responses = responses
+
+# Submit interview and generate final insights
 if st.button("Submit Interview"):
     insights_summary = "\n".join([response[1] for response in st.session_state.responses if response[1]])
     prompt = f"""