Update app.py

app.py

@@ -17,22 +17,32 @@ def start_interview(interview_type, doc_file, name, num_questions):
             chatbot: gr.update(value=[[None, "Please select an interview type and upload a document to begin."]]),
             audio_in: gr.update(interactive=False)
         }
+
     doc_text = extract_text_from_document(doc_file.name)
     if "Error" in doc_text or "Unsupported" in doc_text:
         return {
             chatbot: gr.update(value=[[None, f"Error: {doc_text}"]]),
             audio_in: gr.update(interactive=False)
         }
+
     initial_state = {
-        "interview_type": interview_type, "doc_text": doc_text,
-        "name": name if name else "User", "question_count": int(num_questions),
-        "current_question_num": 1, "interview_log": []
+        "interview_type": interview_type,
+        "doc_text": doc_text,
+        "name": name if name else "User",
+        "question_count": int(num_questions),
+        "current_question_num": 1,
+        "interview_log": [],
+        "start_time": time.time()
     }
+
     first_question = generate_question(interview_type, doc_text)
     initial_state["current_question_text"] = first_question
+    
     greeting = f"Hello {initial_state['name']}. We'll go through {int(num_questions)} questions today. Here is your first question:"
+    
     tts_prompt = f"{greeting} {first_question}"
     ai_voice_path = text_to_speech_file(tts_prompt)
+    
     return {
         state: initial_state,
         chatbot: gr.update(value=[[None, f"{greeting}\n\n{first_question}"]]),
@@ -52,11 +62,14 @@ def handle_interview_turn(user_audio, chatbot_history, current_state):
         "answer": user_answer_text,
         "evaluation": evaluation_text
     })
+
     if current_state["current_question_num"] >= current_state["question_count"]:
         end_message = "This concludes the interview. Generating your final report now."
         chatbot_history.append([None, end_message])
+        
         pdf_path = generate_pdf_file(current_state)
         ai_voice_path = text_to_speech_file(end_message)
+        
         yield {
             chatbot: chatbot_history,
             audio_out: gr.update(value=ai_voice_path, autoplay=True),
@@ -66,10 +79,13 @@ def handle_interview_turn(user_audio, chatbot_history, current_state):
         current_state["current_question_num"] += 1
         next_question = generate_question(current_state["interview_type"], current_state["doc_text"])
         current_state["current_question_text"] = next_question
+        
         q_num = current_state["current_question_num"]
         transition_message = f"Thank you. Here is question {q_num}:\n\n{next_question}"
         chatbot_history.append([None, transition_message])
+        
         ai_voice_path = text_to_speech_file(transition_message)
+
         yield {
             state: current_state,
             chatbot: chatbot_history,
@@ -78,8 +94,15 @@ def handle_interview_turn(user_audio, chatbot_history, current_state):
         }
 
 def generate_pdf_file(state):
-    final_data = { "name": state["name"], "type": state["interview_type"], "q_and_a": state["interview_log"] }
-    file_name = f"Report_{state['name']}_{datetime.datetime.now().strftime('%Y-%m-%d')}.pdf"
+    # This function now correctly creates the dictionary for the report generator
+    total_duration_minutes = (time.time() - state.get("start_time", time.time())) / 60
+    final_data = { 
+        "name": state.get("name", "N/A"), 
+        "type": state.get("interview_type", "N/A"), 
+        "duration": total_duration_minutes,
+        "q_and_a": state.get("interview_log", [])
+    }
+    file_name = f"Report_{state.get('name', 'User')}_{datetime.datetime.now().strftime('%Y-%m-%d')}.pdf"
     file_path = os.path.join(config.REPORT_FOLDER, file_name)
     generate_pdf_report(final_data, file_path)
     return file_path
@@ -95,6 +118,7 @@ with gr.Blocks(theme=gr.themes.Default()) as app:
             num_questions_slider = gr.Slider(minimum=2, maximum=10, value=5, step=1, label="Number of Questions")
             doc_uploader = gr.File(label="Upload Resume/CV")
             start_btn = gr.Button("Start Interview", variant="primary")
+        
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(label="Conversation", height=500)
             audio_in = gr.Audio(sources=["microphone"], type="filepath", label="Record Your Answer", interactive=False)
@@ -106,6 +130,7 @@ with gr.Blocks(theme=gr.themes.Default()) as app:
         inputs=[interview_type_dd, doc_uploader, user_name, num_questions_slider],
         outputs=[state, chatbot, audio_out, audio_in, start_btn]
     )
+    
     audio_in.stop_recording(
         fn=handle_interview_turn,
         inputs=[audio_in, chatbot, state],