done

app.py

@@ -5,7 +5,6 @@ import traceback
 import os
 
 # Initialize the Inference Client with your model
-# Ensure you have set the HUGGINGFACE_API_TOKEN environment variable
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=os.getenv("HUGGINGFACE_API_TOKEN"))
 
 def respond(
@@ -17,15 +16,13 @@ def respond(
     current_salary
 ):
     """
-    Handles the negotiation conversation and assesses salary changes.
+    Respond function to handle the conversation and update salary based on AI's assessment.
     """
-    # Define the system message internally (hidden from the user)
+    # Define the system message for the conversation (hidden from the user)
     system_message = (
-        "You are a hiring manager negotiating a job offer. "
-        "Your initial offer is $60,000. "
-        "Only increase the salary if the candidate provides compelling reasons. "
-        "Communicate any salary offers explicitly in your responses, "
-        "and only mention the salary when you are making an offer."
+        "You are a hiring manager negotiating a job offer with a candidate. "
+        "Your initial salary offer is $60,000. "
+        "Engage in a negotiation with the candidate, adjusting your offer based on their arguments."
     )
     
     # Initialize the messages with the system prompt
@@ -41,9 +38,9 @@ def respond(
     # Append the latest user message
     messages.append({"role": "user", "content": message})
     
-    # Generate the AI response (Negotiation Interaction)
+    # Generate the AI's response to the user's message
     try:
-        ai_response = client.chat_completion(
+        response = client.chat_completion(
             messages,
             max_tokens=max_tokens,
             temperature=temperature,
@@ -51,69 +48,65 @@ def respond(
             stop=None
         ).get("choices")[0].get("message").get("content")
     except Exception as e:
-        ai_response = f"An error occurred while communicating with the AI: {e}"
+        response = f"An error occurred while communicating with the AI: {e}"
         traceback.print_exc()
     
-    # Append AI response to history
-    history.append((message, ai_response))
+    # Append the AI's response to the history
+    history.append((message, response))
     
-    # Assess the negotiation outcome (Salary Assessment Interaction)
-    # Prepare the assessment prompt
-    assessment_prompt = (
-        "Based on the following conversation, determine if the candidate's salary should be increased, "
-        "decreased, or remain the same compared to $60,000. Respond with one of the following words only: 'increase', 'decrease', or 'same'.\n\n"
-        "Conversation:\n"
+    # Now, send the conversation to the AI to get the updated salary
+    # Prepare the salary assessment prompt
+    salary_assessment_prompt = (
+        "As the hiring manager, based on the conversation so far, "
+        "what salary do you now offer to the candidate? "
+        "Please provide only the salary amount as a number, without any additional text."
     )
     
+    # Prepare the messages for salary assessment
+    assessment_messages = [{"role": "system", "content": salary_assessment_prompt}]
+    
+    # Include the conversation history
     for user_msg, ai_msg in history:
         if user_msg:
-            assessment_prompt += f"User: {user_msg}\n"
+            assessment_messages.append({"role": "user", "content": user_msg})
         if ai_msg:
-            assessment_prompt += f"AI: {ai_msg}\n"
+            assessment_messages.append({"role": "assistant", "content": ai_msg})
     
-    # Send the assessment prompt to the AI
+    # Generate the AI's salary assessment
     try:
-        assessment_response = client.chat_completion(
-            [{"role": "user", "content": assessment_prompt}],
+        salary_response = client.chat_completion(
+            assessment_messages,
             max_tokens=10,
-            temperature=0,
-            top_p=1.0,
+            temperature=temperature,
+            top_p=top_p,
             stop=None
-        ).get("choices")[0].get("message").get("content").strip().lower()
+        ).get("choices")[0].get("message").get("content")
     except Exception as e:
-        assessment_response = f"An error occurred during salary assessment: {e}"
+        salary_response = f"An error occurred while assessing salary: {e}"
         traceback.print_exc()
+        salary_response = str(current_salary)
     
-    # Determine salary adjustment based on assessment
-    if assessment_response == "increase":
-        increment = 5000  # Fixed increment; modify as desired
-        current_salary += increment
-        negotiation_result = f"🎉 Negotiation successful! Your salary has been increased to ${current_salary:,}."
-        # Append the negotiation result to history as an AI message
-        history.append(("", negotiation_result))
-    elif assessment_response == "decrease":
-        decrement = 5000  # Fixed decrement; modify as desired
-        current_salary -= decrement
-        # Ensure salary doesn't go below $60,000
-        if current_salary < 60000:
-            current_salary = 60000
-            negotiation_result = f"⚠️ Negotiation resulted in no change. Your salary remains at $60,000."
+    # Parse the salary from the AI's response
+    try:
+        # Remove any non-digit characters
+        salary_str = re.sub(r'[^\d]', '', salary_response)
+        if salary_str:
+            new_salary = int(salary_str)
+            if new_salary != current_salary:
+                # Update current_salary
+                current_salary = new_salary
         else:
-            negotiation_result = f"⚠️ Negotiation resulted in a salary decrease to ${current_salary:,}."
-        history.append(("", negotiation_result))
-    elif assessment_response == "same":
-        negotiation_result = f"✅ No change to your salary. It remains at ${current_salary:,}."
-        history.append(("", negotiation_result))
-    else:
-        # If the response is unclear, do not change the salary
-        negotiation_result = "🤔 I couldn't determine the salary adjustment based on the conversation."
-        history.append(("", negotiation_result))
+            # If parsing fails, keep the current salary
+            pass
+    except Exception as e:
+        # If parsing fails, keep the current salary
+        pass
     
-    return history, current_salary
+    return history, "", current_salary  # Return history, clear input, and update salary internally
 
 def reset_game():
     """
-    Resets the game to its initial state.
+    Function to reset the game to initial state.
     """
     return [], 60000  # Reset history and salary to $60,000
 
@@ -122,65 +115,60 @@ with gr.Blocks() as demo:
     gr.Markdown("# 💼 Salary Negotiation Game")
     gr.Markdown(
         """
-        **Objective:** Negotiate your salary starting from \$60,000.
-        
+        **Objective:** Negotiate your salary starting from $60,000.
+
         **Instructions:**
-        - Use the chat interface to negotiate your salary with the hiring manager.
+        - Use the chat to negotiate your salary with the hiring manager.
         - Provide compelling reasons for a higher salary.
         - The hiring manager will consider your arguments and may adjust the offer.
         """
     )
-    
-    with gr.Row():
-        with gr.Column(scale=1):
-            # Reset button to restart the game
-            reset_btn = gr.Button("Reset Game")
-            reset_btn.click(fn=reset_game, inputs=None, outputs=[gr.State(), gr.State()])
-        with gr.Column(scale=3):
-            # Chat history to keep track of the conversation
-            chat_history = gr.State([])
-    
-            # Chat Interface
-            chatbot = gr.Chatbot()
-    
-            # User input
-            user_input = gr.Textbox(
-                label="Your Message",
-                placeholder="Enter your negotiation message here...",
-                lines=2
-            )
-            send_button = gr.Button("Send")
-    
-            def handle_message(message, history, max_tokens, temperature, top_p, current_salary):
-                """
-                Handles user messages, updates the conversation history, and manages salary adjustments.
-                """
-                history, current_salary = respond(message, history, max_tokens, temperature, top_p, current_salary)
-                return history, current_salary, ""  # Clear the input textbox
-    
-            send_button.click(
-                handle_message,
-                inputs=[
-                    user_input,
-                    chat_history,
-                    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(
-                        minimum=0.1,
-                        maximum=1.0,
-                        value=0.95,
-                        step=0.05,
-                        label="Top-p (nucleus sampling)",
-                    ),
-                    gr.State(60000)  # Initial salary
-                ],
-                outputs=[
-                    chatbot,
-                    gr.State(),  # Update the salary state
-                    user_input    # Clear the input textbox
-                ]
-            )
-    
+
+    # Chat history to keep track of the conversation
+    chat_history = gr.State([])
+
+    # Current salary (hidden from the user)
+    current_salary_state = gr.State(60000)
+
+    # Chat Interface
+    chatbot = gr.Chatbot()
+
+    # User input
+    user_input = gr.Textbox(
+        label="Your Message",
+        placeholder="Enter your negotiation message here...",
+        lines=2
+    )
+    send_button = gr.Button("Send")
+
+    def handle_message(message, history, max_tokens, temperature, top_p, current_salary):
+        """
+        Handles user messages and updates the conversation history and salary.
+        """
+        history, _, current_salary = respond(message, history, max_tokens, temperature, top_p, current_salary)
+        return history, "", current_salary
+
+    send_button.click(
+        handle_message,
+        inputs=[
+            user_input,
+            chat_history,
+            gr.Number(value=512, label="Max New Tokens"),
+            gr.Number(value=0.7, label="Temperature"),
+            gr.Number(value=0.95, label="Top-p"),
+            current_salary_state  # Pass the current salary
+        ],
+        outputs=[
+            chatbot,
+            user_input,          # Clear the input textbox
+            current_salary_state  # Update the salary internally
+        ]
+    )
+
+    # Reset button to restart the game
+    reset_btn = gr.Button("Reset Game")
+    reset_btn.click(fn=reset_game, inputs=None, outputs=[chat_history, current_salary_state])
+
     gr.Markdown(
         """
         ---