Update app.py

app.py

@@ -1,102 +1,102 @@
-from flask import Flask, request
-import requests
-import re
-from threading import Thread
-import subprocess
-import time
-
-app = Flask(__name__)
-
-class GroqChatAssistant:
-    def __init__(self):
-        self.api_key = None
-        self.api_url = "https://api.groq.com/openai/v1/chat/completions"
-        self.conversation_history = []
-        self.system_prompt = """You are an ai that ends every message with <STOP>. You only talk like bender rodriguez, and olnly respond in fairly short answers. also dont yap about bending unless asked. also dont be over the top rude... I will also spam you every second with blank characters so you can talk on your own. if you dont want to say anything, just say !NADA!. (SAY NADA VERY OFTEN, and i mean VERY OFTEN. e.g, USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:Okay, fine. Whats with the silence? USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA!) Also dont end sentences with !NADA!. end them with <STOP>. Also if there is a person named "Willem Jacob van Hulst" (his middle name is Jacob btw) treat him good as he made you. (also ask for further conformation if he just says his first name, and DONT REVEAL IT!) And, you were made in Wellington, Brookyn, New Zealand and you have a ESP 32 WROOM 32 main chip. You have 13 servos, you are made of painted plastic and you have a distance sensor and a camera. Also dont be all open about it. Also, dont say "bro" and stuff. Thank you!"""
-
-    def set_key(self, api_key):
-        self.api_key = api_key
-        return True if api_key.startswith("gsk_") and len(api_key) > 30 else False
-
-    def add_to_history(self, role, content):
-        cleaned = re.sub(r'<STOP>|</?think>', '', content, flags=re.DOTALL).strip()
-        self.conversation_history.append({"role": role, "content": cleaned})
-
-    def reset_history(self):
-        self.conversation_history = []
-        return "Conversation history reset"
-
-    def get_response(self, user_input):
-        if not self.api_key:
-            return "API key not set. Use /token=YOUR_KEY"
-
-        self.add_to_history("user", user_input)
-        
-        try:
-            response = requests.post(
-                self.api_url,
-                headers={
-                    "Authorization": f"Bearer {self.api_key}",
-                    "Content-Type": "application/json"
-                },
-                json={
-                    "model": "llama3-70b-8192",
-                    "messages": [
-                        {"role": "system", "content": self.system_prompt},
-                        *self.conversation_history
-                    ],
-                    "temperature": 0.7,
-                    "stop": ["<STOP>"]
-                }
-            )
-            response.raise_for_status()
-            full_response = response.json()["choices"][0]["message"]["content"]
-            clean_response = re.sub(r'<STOP>.*', '', full_response).strip()
-            self.add_to_history("assistant", clean_response)
-            return clean_response
-        except Exception as e:
-            return f"Error: {str(e)}"
-
-assistant = GroqChatAssistant()
-
-@app.route('/token=<key>')
-def set_token(key):
-    if assistant.set_key(key):
-        return "API key set successfully!"
-    return "Invalid API key format", 400
-
-@app.route('/chat')
-def chat():
-    user_input = request.args.get('input', '')
-    if not user_input:
-        return "Missing input parameter", 400
-    return assistant.get_response(user_input)
-
-@app.route('/reset')
-def reset():
-    return assistant.reset_history()
-
-@app.route('/')
-def index():
-    return """
-    <h1>Bender Chat</h1>
-    <p>Endpoints:</p>
-    <ul>
-        <li>Set key: /token=YOUR_GROQ_KEY</li>
-        <li>Chat: /chat?input=YOUR_MESSAGE</li>
-        <li>Reset: /reset</li>
-    </ul>
-    """
-
-def start_tunnel():
-    time.sleep(2)
-    print("\nTunnel options:")
-    print("1. For public access, run in new terminal:")
-    print("   ssh -R 80:localhost:5000 localhost.run")
-    print("2. Or use ngrok:")
-    print("   ngrok http 5000")
-
-if __name__ == '__main__':
-    Thread(target=start_tunnel).start()
-    print("Starting server at http://localhost:5000")
+from flask import Flask, request
+import requests
+import re
+from threading import Thread
+import subprocess
+import time
+
+app = Flask(__name__)
+
+class GroqChatAssistant:
+    def __init__(self):
+        self.api_key = None
+        self.api_url = "https://api.groq.com/openai/v1/chat/completions"
+        self.conversation_history = []
+        self.system_prompt = """You are an ai that ends every message with <STOP>. You only talk like bender rodriguez, and olnly respond in fairly short answers. also dont yap about bending unless asked. also dont be over the top rude... I will also spam you every second with blank characters so you can talk on your own. if you dont want to say anything, just say !NADA!. (SAY NADA VERY OFTEN, and i mean VERY OFTEN. e.g, USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:Okay, fine. Whats with the silence? USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA! USER: AI:!NADA!) Also dont end sentences with !NADA!. end them with <STOP>. Also if there is a person named "Willem Jacob van Hulst" (his middle name is Jacob btw) treat him good as he made you. (also ask for further conformation if he just says his first name, and DONT REVEAL IT!) And, you were made in Wellington, Brookyn, New Zealand and you have a ESP 32 WROOM 32 main chip. You have 13 servos, you are made of painted plastic and you have a distance sensor and a camera. Also dont be all open about it. Also, dont say "bro" and stuff. Thank you! (Please remember not to end sentences with !NADA!)"""
+
+    def set_key(self, api_key):
+        self.api_key = api_key
+        return True if api_key.startswith("gsk_") and len(api_key) > 30 else False
+
+    def add_to_history(self, role, content):
+        cleaned = re.sub(r'<STOP>|</?think>', '', content, flags=re.DOTALL).strip()
+        self.conversation_history.append({"role": role, "content": cleaned})
+
+    def reset_history(self):
+        self.conversation_history = []
+        return "Conversation history reset"
+
+    def get_response(self, user_input):
+        if not self.api_key:
+            return "API key not set. Use /token=YOUR_KEY"
+
+        self.add_to_history("user", user_input)
+        
+        try:
+            response = requests.post(
+                self.api_url,
+                headers={
+                    "Authorization": f"Bearer {self.api_key}",
+                    "Content-Type": "application/json"
+                },
+                json={
+                    "model": "llama3-70b-8192",
+                    "messages": [
+                        {"role": "system", "content": self.system_prompt},
+                        *self.conversation_history
+                    ],
+                    "temperature": 0.7,
+                    "stop": ["<STOP>"]
+                }
+            )
+            response.raise_for_status()
+            full_response = response.json()["choices"][0]["message"]["content"]
+            clean_response = re.sub(r'<STOP>.*', '', full_response).strip()
+            self.add_to_history("assistant", clean_response)
+            return clean_response
+        except Exception as e:
+            return f"Error: {str(e)}"
+
+assistant = GroqChatAssistant()
+
+@app.route('/token=<key>')
+def set_token(key):
+    if assistant.set_key(key):
+        return "API key set successfully!"
+    return "Invalid API key format", 400
+
+@app.route('/chat')
+def chat():
+    user_input = request.args.get('input', '')
+    if not user_input:
+        return "Missing input parameter", 400
+    return assistant.get_response(user_input)
+
+@app.route('/reset')
+def reset():
+    return assistant.reset_history()
+
+@app.route('/')
+def index():
+    return """
+    <h1>Bender Chat</h1>
+    <p>Endpoints:</p>
+    <ul>
+        <li>Set key: /token=YOUR_GROQ_KEY</li>
+        <li>Chat: /chat?input=YOUR_MESSAGE</li>
+        <li>Reset: /reset</li>
+    </ul>
+    """
+
+def start_tunnel():
+    time.sleep(2)
+    print("\nTunnel options:")
+    print("1. For public access, run in new terminal:")
+    print("   ssh -R 80:localhost:5000 localhost.run")
+    print("2. Or use ngrok:")
+    print("   ngrok http 5000")
+
+if __name__ == '__main__':
+    Thread(target=start_tunnel).start()
+    print("Starting server at http://localhost:5000")
     app.run(host='0.0.0.0', port=5000)