Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 import json
 import time
@@ -71,6 +70,18 @@ def transcribe_audio(audio_file_path):
         print(f"Error during audio transcription: {e}")
         return None
 
+def check_api_key():
+    """Checks if the OpenAI API key is valid."""
+    api_key = os.getenv("OPENAI_API_KEY")
+    if not api_key:
+        return "❌ API Key Not Found. Please enter in Admin Panel."
+    try:
+        client = OpenAI(api_key=api_key)
+        client.models.list() # Simple API call to check if the key is working
+        return "✅ API Key Loaded."
+    except Exception as e:
+        return f"❌ API Key Invalid: {e}"
+
 
 def conduct_interview(questions, language="English", history_limit=5):
     """
@@ -79,8 +90,27 @@ def conduct_interview(questions, language="English", history_limit=5):
     """
     start_time = time.time()
     openai_api_key = os.getenv("OPENAI_API_KEY")
+
+    initial_message = (
+        "👋 Hi there, I'm Sarah, your friendly AI HR assistant! "
+        "I'll guide you through a series of interview questions to learn more about you. "
+        "Take your time and answer each question thoughtfully."
+    )
+    final_message_content = (
+        "That wraps up our interview. Thank you for your responses—it's been great learning more about you!"
+        " I will share the feedback with HR Team, and they will reach out to you soon." # added line
+    )
+
     if not openai_api_key:
-        raise RuntimeError("OpenAI API key not found. Please add it to your .env or set it in env variables.")
+        placeholder_message = "⚠️ OpenAI API Key not configured. Please enter your API key in the Admin Panel to start the interview."
+        placeholder_audio_path = convert_text_to_speech(placeholder_message)
+
+        def placeholder_interview_step(user_input, audio_input, history):
+            history.append({"role": "assistant", "content": placeholder_message})
+            return history, "", placeholder_audio_path
+
+        return placeholder_interview_step, initial_message, final_message_content
+
 
     # LangChain-based ChatOpenAI
     chat = ChatOpenAI(
@@ -110,15 +140,6 @@ def conduct_interview(questions, language="English", history_limit=5):
     interview_stage = ["questioning"] # "questioning", "user_questions_prompt", "answering_user_questions", "final_message_stage", "finished"
     user_questions_asked = [False] # Flag to track if "Do you have any questions?" has been asked
 
-    initial_message = (
-        "👋 Hi there, I'm Sarah, your friendly AI HR assistant! "
-        "I'll guide you through a series of interview questions to learn more about you. "
-        "Take your time and answer each question thoughtfully."
-    )
-    final_message_content = (
-        "That wraps up our interview. Thank you for your responses—it's been great learning more about you!"
-        " I will share the feedback with HR Team, and they will reach out to you soon." # added line
-    )
 
     updated_system_prompt = system_prompt.replace("{final_message_placeholder}", final_message_content)
 
@@ -135,6 +156,14 @@ def conduct_interview(questions, language="English", history_limit=5):
 
         step_start_time = time.time()
 
+        # Check if API key is configured before proceeding with OpenAI calls
+        if not os.getenv("OPENAI_API_KEY"):
+            api_missing_message = "⚠️ OpenAI API Key not configured. Please enter your API key in the Admin Panel to continue the interview."
+            api_missing_audio_path = convert_text_to_speech(api_missing_message)
+            history.append({"role": "assistant", "content": api_missing_message})
+            return history, "", api_missing_audio_path
+
+
         # If there's audio, transcribe it.
         if audio_input:
             transcript = transcribe_audio(audio_input)
@@ -200,7 +229,7 @@ def conduct_interview(questions, language="English", history_limit=5):
             # Should not reach here as final message is sent directly and stage becomes "finished"
             pass
         elif interview_stage[0] == "finished":
-            return history, "", None # Interview is finished
+            return history, "", None  # Interview is finished
 
 
         if messages: # Proceed only if messages are prepared (not in final_message_stage or finished)
@@ -273,7 +302,7 @@ def conduct_interview(questions, language="English", history_limit=5):
             pass # Should not reach here
 
         elif interview_stage[0] == "finished":
-            return history, "", None # Interview already finished
+            return history, "", None  # Interview already finished
 
         print(f"DEBUG - Interview step time: {time.time() - step_start_time:.2f} seconds")
         return history, "", audio_file_path
@@ -293,11 +322,10 @@ def main():
         print(f"Error reading questions: {e}")
         return
 
-    try:
-        interview_func, initial_message, final_message = conduct_interview(questions)
-    except Exception as e:
-        print(f"Error setting up interview: {e}")
-        return
+    global initial_api_key_status_message # Declare as global to modify
+    initial_api_key_status_message = check_api_key() # Check API key and update status
+
+    interview_func, initial_message, final_message = conduct_interview(questions) # Initialize even if API key is missing
 
     css = """
     .contain { display: flex; flex-direction: column; }
@@ -308,8 +336,6 @@ def main():
     .bot > div > .message { background-color: #f7f7f8 !important }
     """
 
-    initial_api_key_status_message = "API Key Status: Checking..."
-
     # Build Gradio interface
     with gr.Blocks(css=css) as demo:
         gr.Markdown(
@@ -352,14 +378,11 @@ def main():
                 gr.Markdown("*This application does not store your API key. It is used only for this session and is not persisted when you close the app.*")
 
                 def update_api_key(api_key):
-                    os.environ["OPENAI_API_KEY"] = api_key # Caution: Modifying os.environ is session-based
-                    global interview_func, initial_message, final_message # Declare globals to update them
-                    try:
-                        interview_func, initial_message, final_message = conduct_interview(questions) # Re-init interview function
-                        return "✅ API Key Updated and Loaded."
-                    except RuntimeError as e:
-                        return f"❌ API Key Update Failed: {e}"
-
+                    os.environ["OPENAI_API_KEY"] = api_key  # Caution: Modifying os.environ is session-based
+                    global interview_func, initial_message, final_message, questions, initial_api_key_status_message  # Declare globals to update them and questions
+                    initial_api_key_status_message = check_api_key() # Update status immediately after key is entered
+                    interview_func, initial_message, final_message = conduct_interview(questions)  # Re-init interview function
+                    return initial_api_key_status_message # Return status message
 
                 update_api_key_button.click(
                     update_api_key,
@@ -522,7 +545,14 @@ def main():
             Must return a list of {'role':'assistant','content':'...'} messages
             plus empty text for user_input and path for audio_output.
             """
-            nonlocal interview_func, questions # Access questions from the outer scope
+            global interview_func, questions, initial_api_key_status_message # Access global variables
+
+            current_api_key_status = check_api_key() # Check API key status right before starting interview
+            if not current_api_key_status.startswith("✅"): # If API key is not valid
+                error_message = "Please set a valid OpenAI API Key in the Admin Panel before starting the interview."
+                tts_path = convert_text_to_speech(error_message)
+                return [{"role": "assistant", "content": error_message}], "", tts_path
+
             try:
                 questions = read_questions_from_json(QUESTIONS_FILE_PATH) # Reload questions in case file changed
                 interview_func, initial_message, final_message = conduct_interview(questions) # Re-init interview func with new questions
@@ -570,7 +600,7 @@ def main():
             Re-initialize the interview function entirely
             to start from scratch, clearing the Chatbot.
             """
-            nonlocal interview_func, initial_message, final_message, questions # Access questions
+            global interview_func, initial_message, final_message, questions # Access global variables
             interview_func, initial_msg, final_msg = conduct_interview(questions) # Re-init with current questions
             return [], "", None
 
@@ -620,4 +650,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()