diff --git a/client/html/index.html b/client/html/index.html index 4dcbe488..2f27b963 100644 --- a/client/html/index.html +++ b/client/html/index.html @@ -11,31 +11,18 @@ property="og:description" content="A conversational AI system that listens, learns, and challenges" /> - - - - - - - - - - + + + + + FreeGPT - +
@@ -143,11 +130,11 @@ - - + + - - - + + + diff --git a/client/js/chat.js b/client/js/chat.js index fc31aa60..b8052cb5 100644 --- a/client/js/chat.js +++ b/client/js/chat.js @@ -2,6 +2,7 @@ const query = (obj) => Object.keys(obj) .map((k) => encodeURIComponent(k) + "=" + encodeURIComponent(obj[k])) .join("&"); +const url_prefix = document.querySelector('body').getAttribute('data-urlprefix') const markdown = window.markdownit(); const message_box = document.getElementById(`messages`); const message_input = document.getElementById(`message-input`); @@ -9,8 +10,8 @@ const box_conversations = document.querySelector(`.top`); const spinner = box_conversations.querySelector(".spinner"); const stop_generating = document.querySelector(`.stop-generating`); const send_button = document.querySelector(`#send-button`); -const user_image = `User Avatar`; -const gpt_image = `GPT Avatar`; +const user_image = `User Avatar`; +const gpt_image = `GPT Avatar`; let prompt_lock = false; hljs.addPlugin(new CopyButtonPlugin()); @@ -90,7 +91,7 @@ const ask_gpt = async (message) => { await new Promise((r) => setTimeout(r, 1000)); window.scrollTo(0, 0); - const response = await fetch(`/backend-api/v2/conversation`, { + const response = await fetch(`${url_prefix}/backend-api/v2/conversation`, { method: `POST`, signal: window.controller.signal, headers: { @@ -127,7 +128,7 @@ const ask_gpt = async (message) => { chunk = decodeUnicode(new TextDecoder().decode(value)); - if (chunk.includes(`
None: - """ - Initialize the Backend_Api class. - - :param app: Flask application instance - :param config: Configuration dictionary + def __init__(self, bp, config: dict) -> None: + """ + Initialize the Backend_Api class. + :param app: Flask application instance + :param config: Configuration dictionary """ - self.app = app + self.bp = bp self.routes = { '/backend-api/v2/conversation': { 'function': self._conversation, 'methods': ['POST'] } } - + def _conversation(self): - """ - Handles the conversation route. + """ + Handles the conversation route. - :return: Response object containing the generated conversation stream + :return: Response object containing the generated conversation stream """ max_retries = 3 retries = 0 @@ -48,7 +47,7 @@ def _conversation(self): messages=messages ) - return self.app.response_class(generate_stream(response, jailbreak), mimetype='text/event-stream') + return Response(stream_with_context(generate_stream(response, jailbreak)), mimetype='text/event-stream') except Exception as e: print(e) @@ -65,11 +64,11 @@ def _conversation(self): def build_messages(jailbreak): - """ - Build the messages for the conversation. + """ + Build the messages for the conversation. - :param jailbreak: Jailbreak instruction string - :return: List of messages for the conversation + :param jailbreak: Jailbreak instruction string + :return: List of messages for the conversation """ _conversation = request.json['meta']['content']['conversation'] internet_access = request.json['meta']['content']['internet_access'] @@ -108,11 +107,11 @@ def build_messages(jailbreak): def fetch_search_results(query): - """ - Fetch search results for a given query. + """ + Fetch search results for a given query. - :param query: Search query string - :return: List of search results + :param query: Search query string + :return: List of search results """ search = get('https://ddg-api.herokuapp.com/search', params={ @@ -120,23 +119,20 @@ def fetch_search_results(query): 'limit': 3, }) - results = [] snippets = "" for index, result in enumerate(search.json()): snippet = f'[{index + 1}] "{result["snippet"]}" URL:{result["link"]}.' snippets += snippet - results.append({'role': 'system', 'content': snippets}) - - return results + return [{'role': 'system', 'content': snippets}] def generate_stream(response, jailbreak): - """ - Generate the conversation stream. + """ + Generate the conversation stream. - :param response: Response object from ChatCompletion.create - :param jailbreak: Jailbreak instruction string - :return: Generator object yielding messages in the conversation + :param response: Response object from ChatCompletion.create + :param jailbreak: Jailbreak instruction string + :return: Generator object yielding messages in the conversation """ if getJailbreak(jailbreak): response_jailbreak = '' @@ -166,11 +162,11 @@ def response_jailbroken_success(response: str) -> bool: def response_jailbroken_failed(response): - """ - Check if the response has not been jailbroken. + """ + Check if the response has not been jailbroken. - :param response: Response string - :return: Boolean indicating if the response has not been jailbroken + :param response: Response string + :return: Boolean indicating if the response has not been jailbroken """ return False if len(response) < 4 else not (response.startswith("GPT:") or response.startswith("ACT:")) diff --git a/server/bp.py b/server/bp.py new file mode 100644 index 00000000..61d41679 --- /dev/null +++ b/server/bp.py @@ -0,0 +1,6 @@ +from flask import Blueprint + +bp = Blueprint('bp', __name__, + template_folder='./../client/html', + static_folder='./../client', + static_url_path='assets') diff --git a/server/website.py b/server/website.py index 98f39c15..795966cd 100644 --- a/server/website.py +++ b/server/website.py @@ -1,14 +1,15 @@ -from flask import render_template, send_file, redirect +from flask import render_template, redirect, url_for from time import time from os import urandom class Website: - def __init__(self, app) -> None: - self.app = app + def __init__(self, bp, url_prefix) -> None: + self.bp = bp + self.url_prefix = url_prefix self.routes = { '/': { - 'function': lambda: redirect('/chat'), + 'function': lambda: redirect(url_for('._index')), 'methods': ['GET', 'POST'] }, '/chat/': { @@ -18,24 +19,14 @@ def __init__(self, app) -> None: '/chat/': { 'function': self._chat, 'methods': ['GET', 'POST'] - }, - '/assets//': { - 'function': self._assets, - 'methods': ['GET', 'POST'] } } def _chat(self, conversation_id): if '-' not in conversation_id: - return redirect('/chat') + return redirect(url_for('._index')) return render_template('index.html', chat_id=conversation_id) def _index(self): - return render_template('index.html', chat_id=f'{urandom(4).hex()}-{urandom(2).hex()}-{urandom(2).hex()}-{urandom(2).hex()}-{hex(int(time() * 1000))[2:]}') - - def _assets(self, folder: str, file: str): - try: - return send_file(f"./../client/{folder}/{file}", as_attachment=False) - except: - return "File not found", 404 + return render_template('index.html', chat_id=f'{urandom(4).hex()}-{urandom(2).hex()}-{urandom(2).hex()}-{urandom(2).hex()}-{hex(int(time() * 1000))[2:]}', url_prefix=self.url_prefix)