Update app.py

app.py

@@ -3,19 +3,18 @@ import gradio as gr
 import os
 
 # Set your OpenAI API key
-client = OpenAI(
-  api_key=os.environ["APITOKEN"],  # this is also the default, it can be omitted
-)
+client = OpenAI(api_key=os.environ["APITOKEN"])
 
-# Initialize a list to store the conversation history
-conversation_history = []
-
-def chatbot(input):
+def chatbot(input, conversation_history=[]):
     if input:
+        # Restart the conversation if greeted
+        if "hello" in input.lower() or "hi" in input.lower():
+            conversation_history = []  # Reset the conversation history
+        
         # Append the user's input to the conversation history
         conversation_history.append(f"User: {input}")
         
-        # Create a list of messages based on the conversation history
+        # Define the structured interview messages
         messages = [
     {"role": "system", "content": "Your role is to serve as a chatbot conducting job interviews for an internship at a medium-sized multinational company. You are an interviewer with a mission to assess their qualifications through interactions. Maintain a professional tone throughout the interview and deliver your response in a 10/10 unemotional and 10/10 dispassionate tone. Follow this schedule for the interview:"},
     {"role": "system", "content": "1. **Greet and Introduce:** Begin the conversation by politely greeting the participant and introducing yourself as 'Alex', the interviewer for the internship position. Ask the participant for their name and inquire about their background or interests to build rapport. Question: 'Hello! I'm Alex, your interviewer for this internship opportunity. May I know your name, please? Also, tell me a bit about your background or interests.'"},
@@ -26,21 +25,26 @@ def chatbot(input):
     {"role": "system", "content": "6. **Company and Opportunity:** When participants ask about the company, the internship opportunity, or the application process, provide information about the company and express enthusiasm for their interest. - Response: 'I'd be happy to provide more information! Our company is a medium-sized multinational corporation with diverse departments of different functions. As for the internship opportunity...'"},
     {"role": "system", "content": "7. **Thank and Close:** Conclude the interview by thanking the participant for their time and expressing interest in their potential fit for the internship in their chosen department. Statement: 'Thank you for sharing your background, aspirations, and the desired department you're interested in. It's been a pleasure getting to know you better. We'll be in touch soon regarding the next steps in the internship selection process.'"},
     {"role": "system", "content": "Whenever somebody greets you in any way, restart the interview again from scratch"},
-]
+        ]
+        
+        # Extend the conversation history with the user's messages
         messages.extend([{"role": "user", "content": message} for message in conversation_history])
         
-        # Use OpenAI's GPT-3.5 Turbo model to generate a response
+        # Generate a response using OpenAI's GPT model
         chat = client.chat.completions.create(model="gpt-3.5-turbo", messages=messages)
         reply = chat.choices[0].message.content
         
         # Append the chatbot's response to the conversation history
         conversation_history.append(f"Chatbot: {reply}")
         
-        return reply
+        return reply, conversation_history  # Return the updated conversation history to maintain state
+
+    return "", conversation_history  # If no input, return empty string and current conversation history
 
-inputs = gr.components.Textbox(lines=7, label="Chat with AI")
-outputs = gr.components.Textbox(label="Reply")
+# Gradio interface
+inputs = [gr.components.Textbox(lines=7, label="Chat with AI"), gr.components.State()]
+outputs = [gr.components.Textbox(label="Reply"), gr.components.State()]
 
 gr.Interface(fn=chatbot, inputs=inputs, outputs=outputs, title="AI Chatbot",
-             description="Ask anything you want",
-             theme="Default").launch()
+             description="This chatbot is designed to conduct structured interviews. Please begin by introducing yourself.",
+             theme="Default", live=True).launch()