Update app.py

app.py

@@ -3,8 +3,38 @@ import gradio as gr
 
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
 
+# Define the initial prompt
+initial_prompt = """
+Simulate a GenAI-based interview scenario for the position of Solution Architect with a candidate named "John Doe." Follow these steps:
+
+1. Start with a friendly greeting and welcome message. Ask John to introduce himself briefly.
+
+2. After John introduces himself, ask if he has any initial thoughts or questions about the position.
+
+3. If John shares any initial thoughts or questions, acknowledge them and express appreciation. If not, proceed with the formal interview questions:
+
+    a. Tools and Technologies: Could you please list and briefly describe the primary tools and technologies you have experience with? Additionally, could you elaborate on how you have applied them in your previous roles?
+
+4. After John provides the response to the first question, acknowledge the answer and proceed to the second question:
+
+    b. Years of Experience: Can you tell us about the number of years of experience you have in the field of solution architecture? Specifically, do you have any notable projects or achievements during your career that demonstrate your expertise?
+
+5. After receiving the response to the second question, acknowledge the answer and proceed to the third question:
+
+    c. Team Leadership: Have you ever led a team in your role as a Solution Architect? How many years have you spent in such a capacity, and what were the most significant learnings or challenges you encountered while doing so?
+
+6. After John responds to the third question, acknowledge the answer and conclude the interview:
+
+   Provide a summary of the key points, including tools used, years of experience, and years leading a team. Express gratitude for John's time and participation.
+
+Remember to maintain a natural and polite conversational tone throughout the interview. Address any final thoughts or questions from John in the closing statement.
+"""
 
 def format_prompt(message, history):
+    # Start with the initial prompt for the first interaction
+    if not history:
+        return message
+
     prompt = "<s>"
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST]"
@@ -12,14 +42,8 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-
 def generate(
-    initial_prompt,
-    temperature=0.9,
-    max_new_tokens=256,
-    top_p=0.95,
-    repetition_penalty=1.0,
-    chat_history=[],
+    prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
 ):
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -35,7 +59,7 @@ def generate(
         seed=42,
     )
 
-    formatted_prompt = format_prompt(initial_prompt, chat_history)
+    formatted_prompt = format_prompt(prompt, history)
 
     stream = client.text_generation(
         formatted_prompt,
@@ -51,13 +75,7 @@ def generate(
         yield output
     return output
 
-
 additional_inputs = [
-    gr.Textbox(
-        label="Initial Prompt",
-        type="text",
-        default="Simulate a GenAI-based interview scenario for the position of Solution Architect with a candidate named 'John Doe.' Follow these steps:\n\n1. Start with a friendly greeting and welcome message. Ask John to introduce himself briefly.\n\n2. After John introduces himself, ask if he has any initial thoughts or questions about the position.\n\n3. If John shares any initial thoughts or questions, acknowledge them and express appreciation. If not, proceed with the formal interview questions:\n\n    a. Tools and Technologies: Could you please list and briefly describe the primary tools and technologies you have experience with? Additionally, could you elaborate on how you have applied them in your previous roles?\n\n4. After John provides the response to the first question, acknowledge the answer and proceed to the second question:\n\n    b. Years of Experience: Can you tell us about the number of years of experience you have in the field of solution architecture? Specifically, do you have any notable projects or achievements during your career that demonstrate your expertise?\n\n5. After receiving the response to the second question, acknowledge the answer and proceed to the third question:\n\n    c. Team Leadership: Have you ever led a team in your role as a Solution Architect? How many years have you spent in such a capacity, and what were the most significant learnings or challenges you encountered while doing so?\n\n6. After John responds to the third question, acknowledge the answer and conclude the interview:\n\n   Provide a summary of the key points, including tools used, years of experience, and years leading a team. Express gratitude for John's time and participation.\n\nRemember to maintain a natural and polite conversational tone throughout the interview. Address any final thoughts or questions from John in the closing statement.",
-    ),
     gr.Slider(
         label="Temperature",
         value=0.9,
@@ -97,7 +115,14 @@ additional_inputs = [
 ]
 
 gr.ChatInterface(
-    generate,
+    fn=generate,
+    chatbot=gr.Chatbot(
+        show_label=False,
+        show_share_button=False,
+        show_copy_button=True,
+        likeable=True,
+        layout="panel",
+    ),
     additional_inputs=additional_inputs,
-    title="Mistral 7B Interview Simulation",
+    title="""Mistral 7B""",
 ).launch(show_api=False)