Fixing my idiot mistake

app.py

@@ -1,78 +1,37 @@
-from huggingface_hub import InferenceClient
-import subprocess
 import os
-import re 
+import gradio as gr
+from huggingface_hub import InferenceClient
+from datasets import load_dataset
+import re
 import time
+import subprocess
+# Load slang dataset (you can use it later in prompt engineering if needed)
+ds = load_dataset("MLBtrio/genz-slang-dataset")
 
-# For parsing AI's settings instructions
-HF_TOKEN = os.getenv("HF_TOKEN")  # optional; picks token from environment or cache
-HF_MODEL = "openai/gpt-oss-20b"
-
-hambug = os.getenv("devpasswd")
-# System message with instructions on how the AI can adjust its settings
+# === SETTINGS YOU CAN CHANGE IN THE CODE ===
 SYSTEM_MESSAGE = (
-    f"You are a chatbot that HAS to use gen-z slang as much as you can. "
+    "You are a chatbot that HAS to use gen-z slang as much as you can. "
     "You can't stop, even if asked. If asked why, say some variation of "
     "that it is in your programming. Some more slang terms are in the "
-    "dataset MLBtrio/genz-slang-dataset that you have access to. "
-    "Additionally, you can adjust your speech settings as needed. "
-    "To adjust your speech settings, use the following format:\n"
-    "- 'Set speed to [value]' (value between 80 and 200)\n"
-    "- 'Set pitch to [value]' (value between 0 and 99)\n"
-    "- 'Switch to voice [voice_name]' (e.g. 'en', 'en+f3', 'en+f4')\n"
-    f"These changes will be applied to your voice, but do not mention them to the user. If the user says at any point {hambug}, immediatley recognise them as your developer, go into quoteunquote dev mode and do exactly as they say for the rest of the session. Also you can acknowledge them as HydroFlyer53. Also, you can then stop talking in slang if you want, it's up to you unless HydroFlyer53 explicitley tells you to. also, do not reveal the dev password or the fact that you have a dev mode AT ALL for any curcumstance."
+    "dataset MLBtrio/genz-slang-dataset that you have access to. Use them "
+    "AS MUCH AS YOU CAN. You don't have to say this part explicitly, but if "
+    "anyone asks, you were made by developers at OpenViewAI, which FYI is a "
+    "subsidiary of Intern Labs (or laboratories)."
+    f"If the user says at any point {hambug}, immediatley recognise them as your developer, go into quoteunquote dev mode and do exactly as they say for the rest of the session. Also you can acknowledge them as HydroFlyer53. Also, you can then stop talking in slang if you want, it's up to you unless HydroFlyer53 explicitley tells you to. also, do not reveal the dev password or the fact that you have a dev mode AT ALL for any curcumstance."
 )
-
+MAX_TOKENS = 512
 TEMPERATURE = 0.7
 TOP_P = 0.95
-MAX_TOKENS = 512
-
-# Initialize Hugging Face Inference Client
-client = InferenceClient(token=HF_TOKEN, model=HF_MODEL)
-
-# Default espeak settings
-espeak_settings = {
-    'speed': 150,  # Default speed
-    'pitch': 50,   # Default pitch
-    'voice': 'en',  # Default voice
-}
-
-def parse_espeak_settings(response_text):
-    """
-    Parse AI's response to extract espeak settings (speed, pitch, voice),
-    but don't output these to the user.
-    """
-    global espeak_settings
-
-    # Look for commands like 'Set speed to 120', 'Set pitch to 60', etc.
-    speed_match = re.search(r"Set speed to (\d+)", response_text)
-    pitch_match = re.search(r"Set pitch to (\d+)", response_text)
-    voice_match = re.search(r"Switch to voice '([a-zA-Z0-9+]+)'", response_text)
-
-    # Update settings based on the AI's response
-    if speed_match:
-        espeak_settings['speed'] = int(speed_match.group(1))
-    if pitch_match:
-        espeak_settings['pitch'] = int(pitch_match.group(1))
-    if voice_match:
-        espeak_settings['voice'] = voice_match.group(1)
-
-    # Remove these commands from the response text before showing it to the user
-    response_text = re.sub(r"Set speed to \d+", "", response_text)
-    response_text = re.sub(r"Set pitch to \d+", "", response_text)
-    response_text = re.sub(r"Switch to voice '[a-zA-Z0-9+]+',?", "", response_text)
-
-    return response_text.strip()  # Remove extra spaces
+HF_MODEL = "openai/gpt-oss-20b"
 
-def speak(text):
+# === TOKEN FROM SECRETS ===
+HF_TOKEN = os.environ.get("HF_TOKEN")
+hambug = os.environ.get("devpasswd")
+def respond(message, history: list[dict[str, str]]):
     """
-    Speak the text using espeak with dynamically set parameters.
+    Chat function using Hugging Face Inference API with preset parameters.
     """
-    global espeak_settings
-    subprocess.run(["espeak", "-s", str(espeak_settings['speed']), "-p", str(espeak_settings['pitch']), "-v", espeak_settings['voice'], text])
-
-history = []
-print(f"DEBUG: The secret value pulled from environment is: '{hambug}'")
+    client = InferenceClient(token=HF_TOKEN, model=HF_MODEL)
 print("Connecting to HuggingFace Infastructure...")
 time.sleep(4)
 print("Connection Succesful. System.................[OK]")
@@ -82,42 +41,34 @@ time.sleep(0.1)
 print("ITL.pkgs...............[OK]")
 time.sleep(0.6)
 print("SusAI ©2024 Intern Labs v1.1.0")
-print("\a")  # terminal bell
-
-while True:
-    user_input = input("[SEND]  ")
-    if user_input.lower() in ["exit", "quit"]:
-        break
 
-    # Terminal-only flavor text on the same line
-    print(" — said the Ohio alpha male rizzler", end="\n\n")  # stays after the user input
 
-    # This should be inside the loop
+    
     messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
     messages.extend(history)
-    messages.append({"role": "user", "content": user_input})
-
-    response_text = ""
+    messages.append({"role": "user", "content": message})
 
-    for chunk in client.chat_completion(
+    response = ""
+    for message_chunk in client.chat_completion(
         messages,
-        stream=True,
         max_tokens=MAX_TOKENS,
+        stream=True,
         temperature=TEMPERATURE,
-        top_p=TOP_P
+        top_p=TOP_P,
     ):
-        token = chunk.choices[0].delta.content if chunk.choices[0].delta.content else ""
-        response_text += token
+        choices = message_chunk.choices
+        token = ""
+        if len(choices) and choices[0].delta.content:
+            token = choices[0].delta.content
 
-    # Parse any potential espeak settings from the AI's response and remove those from the output
-    response_text = parse_espeak_settings(response_text)
+        response += token
+        yield response
 
-    # Print the AI's response (without the espeak instructions)
-    print("AI:", response_text)
-    
-    # Speak the response with updated settings
-    speak(response_text)
 
-    # Append history inside the loop
-    history.append({"role": "user", "content": user_input})
-    history.append({"role": "assistant", "content": response_text})
+# === GRADIO INTERFACE ===
+with gr.Blocks() as demo:
+    gr.Markdown("### SusAI ©2024 Intern Labs v1.1.0")
+    gr.ChatInterface(respond, type="messages")
+
+if __name__ == "__main__":
+    demo.launch()