Update app.py

app.py

@@ -144,7 +144,7 @@ def get_styles():
     with open("style.json", "r") as style_file:
         styles = json.load(style_file)
 
-    return jsonify({"styles": [style["name"] for style in styles]})
+    return jsonify({'status': 'success', "styles": [style["name"] for style in styles]})
 
     
 @app.route('/upload/image', methods=['GET'])
@@ -259,10 +259,12 @@ genai.configure(api_key="AIzaSyBPIdkEyVTDZnmXrBi4ykf0sOfkbOvxAzo")
 
 DEFAULT_MODELS = ["gemini-1.0-pro", "gemini-1.0-pro-001"]
 
-@app.route('/gemini', methods=['GET'])
+@app.route('/gemini', methods=['POST'])
 def gemini():
-    prompt = request.args.get('prompt')
-    model_name = request.args.get('model')
+    data = request.json
+    prompt = data.get('prompt')
+    model_name = data.get('model')
+    messages = data.get('messages', [])
 
     if not prompt:
         return jsonify({'error': 'Prompt parameter is required'}), 400
@@ -295,16 +297,55 @@ def gemini():
         )
 
         # Start the conversation and generate response
-        convo = model.start_chat(history=[])
+        convo = model.start_chat(history=messages)
         convo.send_message(prompt)
         response = convo.last.text
 
-        return jsonify({'status': 'true', 'response': response})
+        return jsonify({'status': 'success', 'response': response})
 
     except Exception as e:
         error_message = str(e)
         app.logger.error("Failed to generate content: %s", error_message)
-        return jsonify({'status': 'false', 'error': 'Failed to generate content.'}), 500
+        return jsonify({'status': 'error', 'error': 'Failed to generate content.'}), 500
+
+
+
+TOKEN_MESSAGES_TMP = {}
+
+def generate_chat_id():
+    """Generate a random chat ID."""
+    while True:
+        chat_id = "Kastg_" + ''.join(random.choices(string.ascii_letters + string.digits, k=random.randint(15, 30)))
+        if len(chat_id) > 7:
+            return chat_id
+
+@app.route('/tmp/chat', methods=['POST'])
+def chat_tmp():
+    data = request.json
+    messages = data.get('messages')
+    chat_id = data.get('chat-id')
+
+    if not messages:
+        return jsonify({'status': 'error', 'error': 'Messages parameter is required'}), 400
+
+    if not chat_id:
+        # Generate a new chat ID
+        chat_id = generate_chat_id()
+    elif not chat_id.startswith('Kastg_'):
+        return jsonify({'status': 'error', 'error': 'Chat ID must start with "Kastg_"'}), 400
+    elif len(chat_id) <= 13:
+        return jsonify({'status': 'error', 'error': 'Chat ID must have more than 7 characters after Kastg_'}), 400
+
+    # Save messages under chat ID
+    if chat_id not in TOKEN_MESSAGES_TMP:
+        TOKEN_MESSAGES_TMP[chat_id] = []
+
+    for message in messages:
+        TOKEN_MESSAGES_TMP[chat_id].append(message)
+
+    # Return token and saved messages
+    response_data = {'creator': 'api.Kastg.com', 'status': 'success', 'chat-id': chat_id, 'messages': TOKEN_MESSAGES_TMP[chat_id]}
+    return jsonify(response_data)
 
     
 if __name__ == "__main__":