Update app.py

app.py

@@ -348,6 +348,38 @@ def chat_tmp():
     response_data = {'creator': 'api.Kastg.com', 'status': 'success', 'chat-id': chat_id, 'messages': TOKEN_MESSAGES_TMP[chat_id]}
     return jsonify(response_data)
 
+
+@app.route('/messages', methods=['POST'])
+def handle_message():
+    try:
+        # Get the data from the request JSON
+        data = request.json
+        messages = data.get('messages', [])
+        model = data.get('model', 'gpt-4o-mini')
+
+        if not messages:
+            return jsonify({"error": "No messages provided"}), 400
+
+        # Validate the structure of messages
+        for message in messages:
+            if 'role' not in message or 'content' not in message:
+                return jsonify({"error": "Invalid message format"}), 400
+
+        # Use the G4F client to get a response
+        response = client.chat.completions.create(
+            model=model,
+            messages=messages
+        )
+
+        # Extract the response content
+        ai_response = response.choices[0].message.content
+
+        # Return the response as JSON
+        return jsonify({"response": ai_response})
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
     
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)