Bug fixing, Cool Stuff, and IDK

app.py

@@ -1,60 +1,123 @@
-import os
-import gradio as gr
 from huggingface_hub import InferenceClient
-from datasets import load_dataset
+import subprocess
+import os
+import re 
+import time
 
-# 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"
 
-# === SETTINGS YOU CAN CHANGE IN THE CODE ===
+hambug = os.getenv("devpasswd")
+# System message with instructions on how the AI can adjust its settings
 SYSTEM_MESSAGE = (
-    "You are a chatbot that HAS to use gen-z slang as much as you can. "
+    f"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. 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)."
+    "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."
 )
-MAX_TOKENS = 512
+
 TEMPERATURE = 0.7
 TOP_P = 0.95
-HF_MODEL = "openai/gpt-oss-20b"
+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)
 
-# === TOKEN FROM SECRETS ===
-HF_TOKEN = os.environ.get("HF_TOKEN")
+    # 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)
 
-def respond(message, history: list[dict[str, str]]):
+    return response_text.strip()  # Remove extra spaces
+
+def speak(text):
     """
-    Chat function using Hugging Face Inference API with preset parameters.
+    Speak the text using espeak with dynamically set parameters.
     """
-    client = InferenceClient(token=HF_TOKEN, model=HF_MODEL)
+    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}'")
+print("Connecting to HuggingFace Infastructure...")
+time.sleep(4)
+print("Connection Succesful. System.................[OK]")
+time.sleep(0.1)
+print("API...............[RECEIVING]")
+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": message})
+    messages.append({"role": "user", "content": user_input})
+
+    response_text = ""
 
-    response = ""
-    for message_chunk in client.chat_completion(
+    for chunk in client.chat_completion(
         messages,
-        max_tokens=MAX_TOKENS,
         stream=True,
+        max_tokens=MAX_TOKENS,
         temperature=TEMPERATURE,
-        top_p=TOP_P,
+        top_p=TOP_P
     ):
-        choices = message_chunk.choices
-        token = ""
-        if len(choices) and choices[0].delta.content:
-            token = choices[0].delta.content
-
-        response += token
-        yield response
+        token = chunk.choices[0].delta.content if chunk.choices[0].delta.content else ""
+        response_text += token
 
+    # Parse any potential espeak settings from the AI's response and remove those from the output
+    response_text = parse_espeak_settings(response_text)
 
-# === GRADIO INTERFACE ===
-with gr.Blocks() as demo:
-    gr.Markdown("### SusAI ©2024 Intern Labs v1.1.0")
-    gr.ChatInterface(respond, type="messages")
+    # Print the AI's response (without the espeak instructions)
+    print("AI:", response_text)
+    
+    # Speak the response with updated settings
+    speak(response_text)
 
-if __name__ == "__main__":
-    demo.launch()
+    # Append history inside the loop
+    history.append({"role": "user", "content": user_input})
+    history.append({"role": "assistant", "content": response_text})