done

app.py

@@ -2,14 +2,11 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import re
 import traceback
+import os
 
 # Initialize the Inference Client with your model
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-
-# Regular expression to validate YouTube URLs (modify if needed)
-YOUTUBE_URL_PATTERN = re.compile(
-    r'(https?://)?(www\.)?(youtube\.com|youtu\.?be)/.+'
-)
+# Ensure you have set the HUGGINGFACE_API_TOKEN environment variable
+client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=os.getenv("HUGGINGFACE_API_TOKEN"))
 
 def respond(
     message,
@@ -42,41 +39,34 @@ def respond(
     # Append the latest user message
     messages.append({"role": "user", "content": message})
     
-    response = ""
-    success = False  # Flag to determine if negotiation was successful
-    
+    # Generate the AI response
     try:
-        # Generate the AI response
-        for msg in client.chat_completion(
+        response = client.chat_completion(
             messages,
             max_tokens=max_tokens,
-            stream=True,
             temperature=temperature,
             top_p=top_p,
-        ):
-            token = msg.choices[0].delta.content
-            response += token
-            yield response
-    
-        # Simple logic to determine negotiation success based on AI's response
-        success_keywords = ["agreed", "accepted", "increase", "raised", "enhanced", "approved", "offer"]
-    
-        # 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
-            success = True
-            yield f"\n\n🎉 Negotiation successful! Your salary has been increased to ${current_salary:,}."
-    
+            stop=None
+        ).get("choices")[0].get("message").get("content")
     except Exception as e:
         response = f"An error occurred while communicating with the AI: {e}"
-        yield response
         traceback.print_exc()
     
-    # Update the salary display if negotiation was successful
-    if success:
-        yield gr.update(value=current_salary)
+    # 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"]
+    
+    # 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
+        negotiation_result = f"🎉 Negotiation successful! Your salary has been increased to ${current_salary:,}."
+        history.append(("", negotiation_result))
+    
+    return history, current_salary
 
 def reset_game():
     """
@@ -114,7 +104,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=3):
             # Chat history to keep track of the conversation
             chat_history = gr.State([])
-
+    
             # Chat Interface
             chatbot = gr.Chatbot()
     
@@ -125,33 +115,13 @@ with gr.Blocks() as demo:
                 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.
                 """
-                generator = respond(message, history, max_tokens, temperature, top_p, current_salary)
-                try:
-                    # Get the initial AI response
-                    response = next(generator)
-                    # Update history
-                    history.append((message, response))
-                    # Process the rest of the generator
-                    try:
-                        while True:
-                            additional_response = next(generator)
-                            response += additional_response
-                            # Check for negotiation success message
-                            if "Negotiation successful" in additional_response:
-                                current_salary = int(re.search(r'\$(\d{1,3}(?:,\d{3})*)', additional_response).group(1).replace(',', ''))
-                            yield (history, current_salary)
-                    except StopIteration:
-                        pass
-                except StopIteration:
-                    pass
-                except Exception as e:
-                    history.append((message, f"An error occurred: {e}"))
-                    yield (history, current_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,
@@ -171,10 +141,11 @@ with gr.Blocks() as demo:
                 ],
                 outputs=[
                     chatbot,
-                    salary_display
+                    salary_display,
+                    user_input  # Clear the input textbox
                 ]
             )
-
+    
     gr.Markdown(
         """
         ---