Update app.py

app.py

@@ -46,7 +46,7 @@ def rank_candidates(candidates):
 def transcribe_video(video_file):
     # Save the uploaded file to a temporary location
     with NamedTemporaryFile(delete=False, suffix=".mp4") as temp_video_file:
-        temp_video_file.write(video_file.read())  # Write video file to the temp file
+        temp_video_file.write(video_file.read())  
         temp_video_path = temp_video_file.name
 
     # Load the video and extract audio
@@ -54,28 +54,43 @@ def transcribe_video(video_file):
     audio_file = "audio.wav"
     video.audio.write_audiofile(audio_file)
     
-    # Perform transcription with Whisper
-    whisper_model = whisper.load_model("base")  # Ensure model is loaded here
+    whisper_model = whisper.load_model("base")  
     transcription = whisper_model.transcribe(audio_file)
 
-    # Clean up temporary files
     os.remove(audio_file)
     os.remove(temp_video_path)
 
     return transcription['text']
 
-# Streamlit app
+
 st.title("AI Role Candidate Screening via Video Interview")
 
-# Input for the number of candidates
+default_videos = ["Unlocking AI_ Insights from a Machine Learning Engineer.mp4", "Navigating Ethical Challenges in AI.mp4"]
+
+
 num_candidates = st.number_input("Enter the number of candidates:", min_value=1, max_value=10, value=1)
 
 mock_interviews = []
 for i in range(num_candidates):
-    video_file = st.file_uploader(f"Upload interview video for Candidate {i+1}:", type=["mp4", "mov", "avi"], key=f"video_{i}")
+    st.write(f"### Candidate {i+1}")
+    
+    # Provide options to either upload a video or use a default video
+    use_default = st.checkbox(f"Use default video for Candidate {i+1}?", key=f"default_{i}")
+    
+    if use_default:
+        video_file_path = default_videos[i % len(default_videos)]  # Cycle through default videos
+        st.write(f"Using default video: {video_file_path}")
+        video_file = open(video_file_path, "rb")  # Open the default video file as binary
+    else:
+        video_file = st.file_uploader(f"Upload interview video for Candidate {i+1}:", type=["mp4", "mov", "avi"], key=f"video_{i}")
+
     if video_file:
         st.write(f"Processing video for Candidate {i+1}...")
-        transcription = transcribe_video(video_file)
+        if not use_default:
+            transcription = transcribe_video(video_file)  # For uploaded videos
+        else:
+            transcription = transcribe_video(video_file)  # For default videos
+        
         st.write(f"Transcript for Candidate {i+1}: {transcription}")
         mock_interviews.append({"name": f"Candidate {i+1}", "response": transcription})
 
@@ -100,4 +115,4 @@ if st.button('Analyze Responses'):
             st.write(f"Average Score: {candidate['avg_score']:.2f}")
             st.write(f"Scores: {candidate['scores']}")
     else:
-        st.write("Please upload videos for all candidates.")
+        st.write("Please upload videos for all candidates.")