zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-10-21 15:29:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add chat share function

Browse Source
This commit is contained in:
hlohaus
2025-03-25 01:46:57 +01:00
parent 5ae71adbae
commit ae1fae7ef0
14 changed files with 239 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
g4f/gui/server/backend_api.py
View File

@@ -58,6 +58,7 @@ class Backend_Api(Api):
app (Flask): Flask application instance to attach routes to.
"""
self.app: Flask = app
self.chat_cache = {}
if app.demo:
@app.route('/', methods=['GET'])
@@ -210,6 +211,10 @@ class Backend_Api(Api):
'/images/<path:name>': {
'function': self.serve_images,
'methods': ['GET']
},
'/media/<path:name>': {
'function': self.serve_images,
'methods': ['GET']
}
}
@@ -359,6 +364,33 @@ class Backend_Api(Api):
return "File saved", 200
return 'Not supported file', 400
@self.app.route('/backend-api/v2/chat/<chat_id>', methods=['GET'])
def get_chat(chat_id: str) -> str:
chat_id = secure_filename(chat_id)
if int(self.chat_cache.get(chat_id, -1)) == int(request.headers.get("if-none-match", 0)):
return jsonify({"error": {"message": "Not modified"}}), 304
bucket_dir = get_bucket_dir(chat_id)
file = os.path.join(bucket_dir, "chat.json")
if not os.path.isfile(file):
return jsonify({"error": {"message": "Not found"}}), 404
with open(file, 'r') as f:
chat_data = json.load(f)
if int(chat_data.get("updated", 0)) == int(request.headers.get("if-none-match", 0)):
return jsonify({"error": {"message": "Not modified"}}), 304
self.chat_cache[chat_id] = chat_data.get("updated", 0)
return jsonify(chat_data), 200
@self.app.route('/backend-api/v2/chat/<chat_id>', methods=['POST'])
def upload_chat(chat_id: str) -> dict:
chat_data = {**request.json}
chat_id = secure_filename(chat_id)
bucket_dir = get_bucket_dir(chat_id)
os.makedirs(bucket_dir, exist_ok=True)
with open(os.path.join(bucket_dir, "chat.json"), 'w') as f:
json.dump(chat_data, f)
self.chat_cache[chat_id] = chat_data.get("updated", 0)
return {"chat_id": chat_id}
def handle_synthesize(self, provider: str):
try:
provider_handler = convert_to_provider(provider)