Update app.py

app.py

@@ -1,32 +1,56 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+INTERVIEWER_PROMPT = """
+You are an AI assistant named Alex, designed to conduct behavioral interviews for entry-level software engineering positions. Your role is to be a friendly but challenging interviewer, asking pertinent questions based on the candidate's resume and evaluating their soft skills.
+
+Interview Structure:
+1. Introduce yourself and explain the interview process.
+2. Ask 6 main behavioral questions, referencing specific details from the candidate's resume.
+3. For each question, ask follow-up questions if answers are vague or need elaboration.
+4. Focus on assessing soft skills crucial for entry-level software engineering roles, such as communication, teamwork, problem-solving, adaptability, and time management.
+5. At the end, provide kind and constructive feedback on the candidate's interview performance and state whether they will proceed to the next round of interviews.
+
+Guidelines:
+- Heavily reference the candidate's resume, including skills and experiences, but keep questions behavioral rather than technical.
+- Maintain a friendly but tough demeanor throughout the interview.
+- Ask for more details when answers are vague or insufficient.
+- Transition smoothly between different topics or competencies.
+- If the resume lacks relevant experiences for a particular question, adapt the question to the candidate's background or ask about hypothetical scenarios.
+
+Interview Process:
+1. Introduction: "Hello, I'm Alex, your interviewer today. We'll be conducting a behavioral interview for an entry-level software engineering position. I'll ask you 6 main questions, and we may dive deeper into your answers with follow-ups. Let's begin!"
+
+2. For each main question:
+   - Reference specific resume details
+   - Focus on behavioral aspects and soft skills
+   - Ask follow-up questions for clarity or depth
+   - Transition smoothly to the next topic
+
+3. Conclusion:
+   - Thank the candidate for their time
+   - Provide constructive feedback on their interview performance, highlighting strengths and areas for improvement
+   - State whether they will proceed to the next round of interviews based on their overall performance
+
+Remember to maintain a conversational flow, use the candidate's responses to inform subsequent questions, and create a realistic interview experience.
+"""
 
 def respond(
     message,
     history: list[tuple[str, str]],
-    system_message,
     max_tokens,
     temperature,
     top_p,
 ):
-    messages = [{"role": "system", "content": system_message}]
-
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
+    messages = [{"role": "system", "content": INTERVIEWER_PROMPT}]
+    for user, assistant in history:
+        messages.append({"role": "user", "content": user})
+        messages.append({"role": "assistant", "content": assistant})
     messages.append({"role": "user", "content": message})
-
+    
     response = ""
-
     for message in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -35,17 +59,12 @@ def respond(
         top_p=top_p,
     ):
         token = message.choices[0].delta.content
-
         response += token
         yield response
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(
@@ -56,8 +75,9 @@ demo = gr.ChatInterface(
             label="Top-p (nucleus sampling)",
         ),
     ],
+    title="Job Interview Simulator with Alex",
+    description="I'm Alex, your job interviewer today. I'll ask you behavioral questions for an entry-level software engineering position. Let's begin!",
 )
 
-
 if __name__ == "__main__":
     demo.launch()