done

app.py

@@ -5,6 +5,7 @@ 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(
@@ -16,13 +17,15 @@ def respond(
     current_salary
 ):
     """
-    Respond function to handle the conversation and negotiate salary.
+    Handles the negotiation conversation and assesses salary changes.
     """
     # Define the system message internally (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."
+        "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."
     )
     
     # Initialize the messages with the system prompt
@@ -38,9 +41,9 @@ def respond(
     # Append the latest user message
     messages.append({"role": "user", "content": message})
     
-    # Generate the AI response
+    # Generate the AI response (Negotiation Interaction)
     try:
-        response = client.chat_completion(
+        ai_response = client.chat_completion(
             messages,
             max_tokens=max_tokens,
             temperature=temperature,
@@ -48,35 +51,69 @@ def respond(
             stop=None
         ).get("choices")[0].get("message").get("content")
     except Exception as e:
-        response = f"An error occurred while communicating with the AI: {e}"
+        ai_response = f"An error occurred while communicating with the AI: {e}"
         traceback.print_exc()
     
     # Append AI response to history
-    history.append((message, response))
-    
-    # Extract salary offers from AI's response
-    # Use regex to find salary amounts in the response
-    salary_pattern = r'\$\s?([\d,]+)'
-    matches = re.findall(salary_pattern, response)
-    offered_salary = None
-    
-    if matches:
-        # Convert extracted salary strings to integers
-        for match in matches:
-            salary_str = match.replace(',', '').strip()
-            try:
-                salary_value = int(salary_str)
-                if salary_value > current_salary:
-                    offered_salary = salary_value
-                    current_salary = offered_salary
-            except ValueError:
-                continue  # Ignore invalid salary strings
+    history.append((message, ai_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"
+    )
+    
+    for user_msg, ai_msg in history:
+        if user_msg:
+            assessment_prompt += f"User: {user_msg}\n"
+        if ai_msg:
+            assessment_prompt += f"AI: {ai_msg}\n"
+    
+    # Send the assessment prompt to the AI
+    try:
+        assessment_response = client.chat_completion(
+            [{"role": "user", "content": assessment_prompt}],
+            max_tokens=10,
+            temperature=0,
+            top_p=1.0,
+            stop=None
+        ).get("choices")[0].get("message").get("content").strip().lower()
+    except Exception as e:
+        assessment_response = f"An error occurred during salary assessment: {e}"
+        traceback.print_exc()
+    
+    # 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."
+        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))
     
     return history, current_salary
 
 def reset_game():
     """
-    Function to reset the game to initial state.
+    Resets the game to its initial state.
     """
     return [], 60000  # Reset history and salary to $60,000
 
@@ -85,57 +122,65 @@ 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 to negotiate your salary with the hiring manager.
+        - Use the chat interface 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.
         """
     )
-
-    # 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 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"),
-            gr.State(60000)  # Initial salary
-        ],
-        outputs=[
-            chatbot,
-            user_input,  # Clear the input textbox
-            gr.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, gr.State(60000)])
-
+    
+    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
+                ]
+            )
+    
     gr.Markdown(
         """
         ---