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(
@@ -22,8 +21,8 @@ def respond(
     # Define the system message internally (hidden from the user)
     system_message = (
         "You are a hiring manager negotiating a job offer. "
-        "Your goal is to offer a fair salary based on the candidate's negotiation skills. "
-        "Start the negotiation with an initial offer of $60,000."
+        "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
@@ -55,16 +54,23 @@ def respond(
     # Append AI response to history
     history.append((message, response))
     
-    # Simple logic to determine negotiation success based on AI's response
-    success_keywords = ["agreed", "accepted", "increase", "raised", "enhanced", "approved", "offer", "agree", "accept"]
+    # 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
     
-    # Check if any success keywords are in the AI's response (case-insensitive)
-    if any(keyword.lower() in response.lower() for keyword in success_keywords):
-        # Increment the salary by a fixed amount
-        increment = 5000  # Fixed increment; modify as desired
-        current_salary += increment
-        # Note: We're NOT appending any message to the chat history here
-        # The salary update is handled separately
+    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
     
     return history, current_salary
 
@@ -72,79 +78,63 @@ def reset_game():
     """
     Function to reset the game to initial state.
     """
-    return [], 60000  # Reset history and salary to 60k
+    return [], 60000  # Reset history and salary to $60,000
 
 # Define the Gradio Blocks layout
 with gr.Blocks() as demo:
     gr.Markdown("# 💼 Salary Negotiation Game")
     gr.Markdown(
         """
-        **Objective:** Negotiate with the employer to increase your salary from \$60,000.
-        
+        **Objective:** Negotiate your salary starting from $60,000.
+
         **Instructions:**
-        - Use the chat interface to negotiate.
-        - If you negotiate successfully, your salary will increase.
-        - Aim to reach the highest salary possible through effective negotiation.
+        - 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):
-            # Display the current salary
-            salary_display = gr.Number(
-                value=60000,
-                label="Current Salary (\$)",
-                interactive=False,
-                precision=0
-            )
-            # Reset button to restart the game
-            reset_btn = gr.Button("Reset Game")
-            reset_btn.click(fn=reset_game, inputs=None, outputs=[gr.State(), salary_display])
+    # Chat history to keep track of the conversation
+    chat_history = 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 and updates the conversation history and salary.
-                """
-                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)",
-                    ),
-                    salary_display
-                ],
-                outputs=[
-                    chatbot,
-                    salary_display,
-                    user_input  # Clear the input textbox
-                ]
-            )
+    # 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)])
 
     gr.Markdown(
         """